Home
last modified time | relevance | path

Searched defs:output_slice_shape (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Donehot_info_test_axis_0.cc140 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
162 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
184 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Donehot_info_test.cc140 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
162 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
184 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dreshape_test.cc107 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
128 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dprelu_test.cc94 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
199 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dtanh_info_test.cc91 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dpow_info_test.cc91 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dgelu_info_test.cc91 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dlog_softmax_info_test.cc92 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dactivation_info_test.cc92 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dtmpidentity_test.cc93 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dmatmul_info_test.cc261 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
287 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
313 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dsoftmax_info_test.cc96 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dl2_normalize_info_test.cc92 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dtranspose_test.cc107 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dreduce_method_test.cc96 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dtensor_add_info_test.cc96 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dreduce_method_info.cc362 Shape output_slice_shape = outputs_slice_shape.at(0); in InferTensorInfo() local
491 Shape output_slice_shape = outputs_slice_shape.at(0); in InferTensorInfo() local
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Doperator_costmodel_test.cc173 Shape output_slice_shape = {8, 32, 8, 8}; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc302 Shape output_slice_shape = output_info.slice_shape(); in GetForwardComputationCost() local
1649 Shape output_slice_shape = outputs[0].slice_shape(); in GetForwardComputationCost() local
1730 Shape output_slice_shape = outputs[0].slice_shape(); in GetForwardComputationCost() local