Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Drange_info.cc53 TensorMap input_tensor_map = {}, output_tensor_map = {MAP_NONE}; in InferTensorMap() local
Dtensordot_info.cc216 TensorMap input_b_tensor_map, output_tensor_map; in InferTensorMapAxesInt() local
236 TensorMap input_b_tensor_map, output_tensor_map, tmp_b_map_index; in InferTensorMapAxesTuple() local
Dmaxpool_info.cc183 TensorMap output_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Dgathernd_info.cc104 TensorMap output_tensor_map(outputs_shape_[0].size(), MAP_NONE); in InferTensorMap() local
Davgpool_info.cc198 TensorMap output_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Ddsd_matmul_info.cc136 TensorMap output_tensor_map; in InferTensorMap() local
Dmoe_finalize_routing_info.cc176 Shape output_tensor_map{2, 0}; in InferTensorMap() local
Drandom_distribute_info.cc82 TensorMap output_tensor_map; in InferTensorMap() local
Dconv3d_info.cc135 TensorMap output_tensor_map = {4, 0, 2, 1, -1}; in InferTensorMap() local
Dtile_info.cc140 TensorMap output_tensor_map; in InferTensorMap() local
Dresizebilinear_v2_info.cc171 TensorMap output_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
Dactivation_info.cc661 Shape input_tensor_map, output_tensor_map; in InferTensorMap() local
770 Shape input_tensor_map, output_tensor_map; in InferTensorMap() local
/third_party/mindspore/mindspore-src/source/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.cc207 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
235 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
263 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
375 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
Dpow_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
Dlog_softmax_info_test.cc113 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
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
Dl2_normalize_info_test.cc113 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

12