Home
last modified time | relevance | path

Searched defs:output_tensor_map (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dresizebilinear_info.cc97 TensorMap output_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Dtensordot_info.cc217 TensorMap input_b_tensor_map, output_tensor_map; in InferTensorMapAxesInt() local
237 TensorMap input_b_tensor_map, output_tensor_map, tmp_b_map_index; in InferTensorMapAxesTuple() local
Drange_info.cc79 TensorMap input_tensor_map = {0}, output_tensor_map = {0}; in InferTensorMap() local
Dmaxpool_info.cc173 TensorMap output_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Dgathernd_info.cc103 TensorMap output_tensor_map(outputs_shape_[0].size(), MAP_NONE); in InferTensorMap() local
Ddsd_matmul_info.cc135 TensorMap output_tensor_map; in InferTensorMap() local
Duniform_real_info.cc78 TensorMap output_tensor_map; in InferTensorMap() local
Dtile_info.cc110 TensorMap output_tensor_map; in InferTensorMap() 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
Dconv2d_info.cc490 TensorMap output_tensor_map = {4, 0, 2, 1}; in InferTensorMap() local
999 TensorMap output_tensor_map = {4, 0, 2, 1}; in InferTensorMap() local
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dreshape_test.cc149 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
170 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dmatmul_info_test.cc184 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
210 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
236 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
339 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dprelu_test.cc118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
223 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dtanh_info_test.cc112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dgelu_info_test.cc112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dpow_info_test.cc112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dtmpidentity_test.cc114 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dactivation_info_test.cc113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dlog_softmax_info_test.cc113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dl2_normalize_info_test.cc113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Donehot_info_test_axis_0.cc118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dsoftmax_info_test.cc117 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dtranspose_test.cc128 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Donehot_info_test.cc118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
Dreduce_method_test.cc117 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local

12