Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/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
Dgelu_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
Dtmpidentity_test.cc93 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
Dlog_softmax_info_test.cc92 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
Dmatmul_info_test.cc291 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
319 Shape output_slice_shape = output_tensor_info.slice_shape(); in TEST_F() local
347 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
Dl2_normalize_info_test.cc92 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
Dreduce_method_test.cc97 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-src/source/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-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc381 Shape output_slice_shape = output_info.slice_shape(); in GetForwardComputationCost() local
1733 Shape output_slice_shape = outputs[0].slice_shape(); in GetForwardComputationCost() local
1814 Shape output_slice_shape = outputs[0].slice_shape(); in GetForwardComputationCost() local