InferShapeFunc
说明
根据输入inTensor信息推导输出outTensor信息。
定义
using InferShapeFunc = std::function<Status(const SVector<TensorDesc> &inTensorDescs, SVector<TensorDesc> &outTensorDescs)>;
父主题: atb/types.h
根据输入inTensor信息推导输出outTensor信息。
using InferShapeFunc = std::function<Status(const SVector<TensorDesc> &inTensorDescs, SVector<TensorDesc> &outTensorDescs)>;