Home
last modified time | relevance | path

Searched defs:input_slice_shape (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Donehot_info_test_axis_0.cc139 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
161 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
183 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Donehot_info_test.cc139 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
161 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
183 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dreshape_test.cc106 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
127 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dprelu_test.cc93 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
198 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dgelu_info_test.cc90 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dpow_info_test.cc90 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dtanh_info_test.cc90 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dlog_softmax_info_test.cc91 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dtmpidentity_test.cc92 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dactivation_info_test.cc91 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dsoftmax_info_test.cc95 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dsoftmax_entropy_loss_info_test.cc95 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dtranspose_test.cc106 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dl2_normalize_info_test.cc91 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
Dreduce_method_test.cc95 Shape input_slice_shape = input_tensor_info.slice_shape(); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Ddropout_do_mask_info.cc181 void SetGenMaskShape(const CNodePtr &cnode, const Shape &input_slice_shape) { in SetGenMaskShape()
244 Shape input_slice_shape = inputs_tensor_info_[0].slice_shape(); in GetDropoutGenMaskReplaceOp() local
Dslice_info.cc221 Shape input_slice_shape = inputs_tensor_info_[0].slice_shape(); in ComputeReplaceGraph() local
Dreduce_method_info.cc361 Shape input_slice_shape = inputs_slice_shape.at(0); in InferTensorInfo() local
490 Shape input_slice_shape = inputs_slice_shape.at(0); in InferTensorInfo() local
/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Doperator_costmodel_test.cc171 Shape input_slice_shape = {8, 32, 8, 8}; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Doperator_costmodel.cc1125 Shape input_slice_shape = input_tensor_info.slice_shape(); in GetBackwardCommCost() local
1453 Shape input_slice_shape = input.slice_shape(); in GetBackwardCommCost() local
1467 Shape input_slice_shape = inputs[0].slice_shape(); in GetForwardComputationCost() local
1482 Shape input_slice_shape = input.slice_shape(); in GetBackwardComputationCost() local