Home
last modified time | relevance | path

Searched defs:output_tensor_layout (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dreduce_method_info.cc290 TensorLayout output_tensor_layout; in InferTensorInfo() local
371 TensorLayout output_tensor_layout; in InferTensorInfo() local
584 TensorLayout input_tensor_layout, output_tensor_layout; in InferTensorInfo() local
Dbias_add_info.cc153 TensorLayout output_tensor_layout; in InferOutputTensorInfo() local
Dgather_info.cc498 TensorLayout output_tensor_layout; in InferTensorInfo() local
859 TensorLayout input_tensor_layout, input_index_layout, output_tensor_layout; in InferTensorInfo() local
1305 TensorLayout output_tensor_layout; in InferTensorInfoNoSplitAxis() local
Drms_norm_info.cc333 TensorLayout output_tensor_layout; in InferOutputLayout() local
Dlayer_norm_info.cc432 TensorLayout output_tensor_layout; in InferOutputLayout() local
Dtranspose_info.cc308 TensorLayout output_tensor_layout; in InferOutputLayout() local
Dmatmul_info.cc651 TensorLayout output_tensor_layout; in InferNDTPOutputLayout() local
748 TensorLayout output_tensor_layout; in InferOutputLayout() local
Darithmetic_info.cc428 TensorLayout output_tensor_layout; in InferOutputLayout() local