Home
last modified time | relevance | path

Searched refs:TensorLayouts (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dunique_info.h48 Status InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout);
Dmatmul_info.h57 Status InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout);
Dreshape_info.h80 Status InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout);
Dget_next_info.h47 Status InferTensorLayout(TensorLayouts *outputs_layout);
Dget_next_info.cc56 Status GetNextInfo::InferTensorLayout(TensorLayouts *outputs_layout) { in InferTensorLayout()
72 TensorLayouts outputs_layout; in InferTensorInfo()
Dreshape_info.cc192 Status ReshapeInfo::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout) { in InferTensorLayout()
262 TensorLayouts inputs_layout, outputs_layout; in InferTensorInfo()
Dmatmul_info.cc314 Status MatMulBase::InferTensorLayout(TensorLayouts *inputs_layout, TensorLayouts *outputs_layout) { in InferTensorLayout()
345 TensorLayouts inputs_layout, outputs_layout; in InferTensorInfo()
Doperator_info.h48 using TensorLayouts = std::vector<TensorLayout>; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2312 TensorLayouts GetLossNodeGradOutputLayout(const LossNodeInfo &node_info) { in GetLossNodeGradOutputLayout()
2313 TensorLayouts ret; in GetLossNodeGradOutputLayout()