下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

set_op_input_tensor_multi_dims

函数原型

def set_op_input_tensor_multi_dims(tensor, input_shape, input_dims)

功能说明

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

使用约束

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

参数说明

参数名

输入/输出

描述

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"

返回值

无。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词