Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dgather_v2_info.cc187 TensorLayout input_tensor_layout, input_index_layout, output_tensor_layout; in InferTensorInfo() local
Dreduce_method_info.cc364 TensorLayout input_tensor_layout, output_tensor_layout; in InferTensorInfo() local
493 TensorLayout input_tensor_layout, output_tensor_layout; in InferTensorInfo() local
Dgather_v2_p_info.cc570 TensorLayout input_tensor_layout, input_index_layout, output_tensor_layout; in InferTensorInfo() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2082 TensorLayout input_tensor_layout; in CreateParameterLayout() local