Searched defs:CreateTensorInfo (Results 1 – 4 of 4) sorted by relevance
255 TensorInfoBasePtr GroupedMatmulInfo::CreateTensorInfo(const Shape &device_matrix, const ShapeBasePt… in CreateTensorInfo() function in mindspore::parallel::GroupedMatmulInfo
550 TensorInfoBasePtr CreateTensorInfo(const Shape &device_matrix, const ShapeBasePtr &inputs_shape, in CreateTensorInfo() function
65 tensor::TensorPtr CreateTensorInfo(const void *data, size_t data_size, const std::vector<int64_t> &… in CreateTensorInfo() function
377 api::TensorPtr CreateTensorInfo(const void *data, size_t data_size, const std::vector<int64_t> &sha… in CreateTensorInfo() function