Home
last modified time | relevance | path

Searched defs:input_tensor_layout (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dreduce_method_info.cc289 TensorLayout input_tensor_layout; in InferTensorInfo() local
370 TensorLayout input_tensor_layout; in InferTensorInfo() local
584 TensorLayout input_tensor_layout, output_tensor_layout; in InferTensorInfo() local
Dgather_info.cc496 TensorLayout input_tensor_layout; in InferTensorInfo() local
859 TensorLayout input_tensor_layout, input_index_layout, output_tensor_layout; in InferTensorInfo() local
1303 TensorLayout input_tensor_layout; in InferTensorInfoNoSplitAxis() local
Dflash_attention_score_info.cc619 auto input_tensor_layout = inputs_tensor_info_[i].tensor_layout(); in InferMirrorOpsByLayout() local
Doperator_info.cc512 auto input_tensor_layout = inputs_tensor_info_[i].tensor_layout(); in InferMirrorOpsByLayout() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc1642 TensorLayout input_tensor_layout; in CreateParameterLayout() local