| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | tensordot_info.cc | 217 TensorMap input_b_tensor_map, output_tensor_map; in InferTensorMapAxesInt() local
223 output_tensor_map = tensor_map_index; in InferTensorMapAxesInt()
224 (void)output_tensor_map.erase( in InferTensorMapAxesInt()
225 …output_tensor_map.begin() + static_cast<different_type>(inputs_shape_[0].size() - IntToSize(axes_i… in InferTensorMapAxesInt()
226 output_tensor_map.begin() + static_cast<different_type>(inputs_shape_[0].size())); in InferTensorMapAxesInt()
229 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMapAxesInt()
237 TensorMap input_b_tensor_map, output_tensor_map, tmp_b_map_index; in InferTensorMapAxesTuple() local
270 output_tensor_map.push_back(tensor_map_index[i]); in InferTensorMapAxesTuple()
274 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMapAxesTuple()
|
| D | dsd_matmul_info.cc | 135 TensorMap output_tensor_map; in InferTensorMap() local
139 output_tensor_map.push_back((int64_t)(outputs_shape_[0].size() - i)); in InferTensorMap()
141 output_tensor_map.push_back((int64_t)(MAP_NONE)); in InferTensorMap()
147 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMap()
|
| D | activation_info.cc | 423 Shape input_tensor_map, output_tensor_map; in InferTensorMap() local
431 output_tensor_map = input_tensor_map; in InferTensorMap()
436 (void)output_tensor_map.insert(output_tensor_map.begin() + positive_axis_, NO_SPLIT_MAP); in InferTensorMap()
437 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMap()
440 << ", and the tensor map of output is " << ShapeToString(output_tensor_map); in InferTensorMap()
553 Shape input_tensor_map, output_tensor_map; in InferTensorMap() local
564 output_tensor_map.push_back(SizeToLong(index)); in InferTensorMap()
569 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMap()
571 << ", and the tensor map of output is " << ShapeToString(output_tensor_map); in InferTensorMap()
|
| D | gathernd_info.cc | 103 TensorMap output_tensor_map(outputs_shape_[0].size(), MAP_NONE); in InferTensorMap() local
105 output_tensor_map[i] = indices_tensor_map[i]; in InferTensorMap()
110 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMap()
|
| D | uniform_real_info.cc | 78 TensorMap output_tensor_map; in InferTensorMap() local
84 output_tensor_map.push_back(SizeToLong(size - i - 1)); in InferTensorMap()
88 (void)outputs_tensor_map_.emplace_back(std::move(output_tensor_map)); in InferTensorMap()
|
| D | range_info.cc | 79 TensorMap input_tensor_map = {0}, output_tensor_map = {0}; in InferTensorMap() local
82 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMap()
|
| D | tile_info.cc | 110 TensorMap output_tensor_map; in InferTensorMap() local
129 output_tensor_map.push_back(size - i - 1); in InferTensorMap()
133 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMap()
|
| D | resizebilinear_info.cc | 97 TensorMap output_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
100 (void)outputs_tensor_map_.emplace_back(std::move(output_tensor_map)); in InferTensorMap()
|
| D | maxpool_info.cc | 173 TensorMap output_tensor_map = {3, 2, 1, 0}; in InferTensorMap() local
176 (void)outputs_tensor_map_.emplace_back(std::move(output_tensor_map)); in InferTensorMap()
|
| D | matmul_info.cc | 262 TensorMap output_tensor_map = tensor_map_index; in InferTensorMap() local
263 …(void)output_tensor_map.erase(output_tensor_map.begin() + static_cast<different_type>(SECOND_FROM_… in InferTensorMap()
304 output_tensor_map = {1, 0}; in InferTensorMap()
310 outputs_tensor_map_.push_back(output_tensor_map); in InferTensorMap()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | reshape_test.cc | 149 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
152 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
170 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
173 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | matmul_info_test.cc | 184 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
188 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
210 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
214 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
236 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
240 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
339 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
343 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | prelu_test.cc | 118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
121 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
223 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
226 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | gelu_info_test.cc | 112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
115 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | log_softmax_info_test.cc | 113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
116 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | pow_info_test.cc | 112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
115 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | tanh_info_test.cc | 112 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
115 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | softmax_info_test.cc | 117 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
120 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | activation_info_test.cc | 113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
116 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | onehot_info_test_axis_0.cc | 118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
121 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | transpose_test.cc | 128 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
131 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | tmpidentity_test.cc | 114 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
117 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | l2_normalize_info_test.cc | 113 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
116 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | onehot_info_test.cc | 118 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
121 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|
| D | reduce_method_test.cc | 117 Map output_tensor_map = output_tensor_info.tensor_layout().origin_tensor_map(); in TEST_F() local
120 ASSERT_EQ(output_tensor_map.array(), output_expect); in TEST_F()
|