Home
last modified time | relevance | path

Searched defs:input_tensor_map (Results 1 – 25 of 31) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dtile_info.cc109 TensorMap input_tensor_map; in InferTensorMap() local
139 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
Dresizebilinear_info.cc96 TensorMap input_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Dprelu_info.cc68 TensorMap input_tensor_map; in InferTensorMap() local
Drange_info.cc79 TensorMap input_tensor_map = {0}, output_tensor_map = {0}; in InferTensorMap() local
Dmaxpool_info.cc172 TensorMap input_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Dgathernd_info.cc94 TensorMap input_tensor_map(inputs_shape_[0].size(), MAP_NONE); // the input can not split in InferTensorMap() local
Dscatter_update_info.cc112 TensorMap input_tensor_map, updates_tensor_map; in InferTensorMap() local
Dbatchnorm_info.cc111 TensorMap input_tensor_map; in InferTensorMap() local
Duniform_real_info.cc77 TensorMap input_tensor_map; in InferTensorMap() local
Dgatherd_info.cc100 Shape input_tensor_map; in InferTensorMap() local
Dstrided_slice_info.cc183 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
Dactivation_info.cc423 Shape input_tensor_map, output_tensor_map; in InferTensorMap() local
553 Shape input_tensor_map, output_tensor_map; in InferTensorMap() local
Dslice_info.cc131 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps() local
Dreduce_method_info.cc285 Shape input_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps() local
310 Shape input_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps() local
Dconv2d_info.cc488 TensorMap input_tensor_map = {4, 3, 2, 1}; in InferTensorMap() local
997 TensorMap input_tensor_map = {4, 3, 2, 1}; in InferTensorMap() local
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dreshape_test.cc148 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
169 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dprelu_test.cc116 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
221 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dtanh_info_test.cc111 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dpow_info_test.cc111 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dgelu_info_test.cc111 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dlog_softmax_info_test.cc112 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dactivation_info_test.cc112 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dtmpidentity_test.cc113 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Donehot_info_test_axis_0.cc117 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dsoftmax_info_test.cc116 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local

12