set_op_input_tensor_multi_dims

用于子图分档功能，指定当前算子的输入shape以及所有分档档位的shape信息。

def set_op_input_tensor_multi_dims(tensor, input_shape, input_dims)

参数名	输入/输出	描述
tensor	输入	目标算子的任意输出tensor。
input_shape	输入	目标算子所有输入tensor的shape，例如：若目标算子有2个输入tensor： "0: -1, -1, 3; 1: 2, 2, -1, 224"
input_dims	输入	动态dim的分档档位信息，每个档位间用";"分隔，档位内dim数量与input_shape中"-1"的数量和顺序一致，例如：若共有3个档位： "480, 480, 112; 960, 960, 224; 1920, 1920, 448"

参数名

输入/输出

描述

tensor

输入

目标算子的任意输出tensor。

input_shape

输入

目标算子所有输入tensor的shape，例如：

若目标算子有2个输入tensor：
"0: -1, -1, 3; 1: 2, 2, -1, 224"

input_dims

输入

动态dim的分档档位信息，每个档位间用";"分隔，档位内dim数量与input_shape中"-1"的数量和顺序一致，例如：

若共有3个档位：
"480, 480, 112; 960, 960, 224; 1920, 1920, 448"

无。

该接口需要配合分档范围划分接口（即subgraph_multi_dims_scope）使用，且目标算子必须为分档范围的输入节点，否则不生效。

父主题： npu_bridge.estimator.npu.util