Home
last modified time | relevance | path

Searched refs:origin_tensor_map (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dmatmul_info_test.cc182 Map mat_a_tensor_map = mat_a_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
183 Map mat_b_tensor_map = mat_b_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
184 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
208 Map mat_a_tensor_map = mat_a_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
209 Map mat_b_tensor_map = mat_b_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
210 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
234 Map mat_a_tensor_map = mat_a_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
235 Map mat_b_tensor_map = mat_b_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
236 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
337 Map mat_a_tensor_map = mat_a_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
[all …]
Dprelu_test.cc116 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
117 Map param_tensor_map = param_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
221 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
222 Map param_tensor_map = param_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
223 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dsoftmax_entropy_loss_info_test.cc124 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
125 Map label_tensor_map = label_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
126 Map output_0_tensor_map = output_0_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
127 Map output_1_tensor_map = output_1_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dreshape_test.cc148 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
149 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
169 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
170 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dtensor_add_info_test.cc118 Map inputa_tensor_map = inputa_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
119 Map inputb_tensor_map = inputb_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
120 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dgelu_info_test.cc111 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dlog_softmax_info_test.cc112 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dpow_info_test.cc111 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dtanh_info_test.cc111 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dsoftmax_info_test.cc116 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
117 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dactivation_info_test.cc112 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Donehot_info_test_axis_0.cc117 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dtranspose_test.cc127 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
128 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dtmpidentity_test.cc113 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
114 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dl2_normalize_info_test.cc112 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Donehot_info_test.cc117 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
Dreduce_method_test.cc116 Map input_tensor_map = input_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
117 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_layout.h67 Map origin_tensor_map() const { return tensor_map_origin_; } in origin_tensor_map() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.cc556 Shape tensor_map = tensor_layout->origin_tensor_map().array(); in CreateGroupForOptShard()