Home
last modified time | relevance | path

Searched refs:inputs_tensor_map_ (Results 1 – 25 of 44) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dgatherd_info.cc106 inputs_tensor_map_.push_back(input_tensor_map); // input in InferTensorMap()
107 inputs_tensor_map_.push_back(input_tensor_map); // index in InferTensorMap()
113 …if (inputs_shape_.empty() || outputs_shape_.empty() || inputs_tensor_map_.empty() || outputs_tenso… in InferTensorInfo()
118 for (size_t i = 0; i < inputs_tensor_map_.size(); ++i) { in InferTensorInfo()
120 …if (input_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[i], inputs_shape_[i]) != SUC… in InferTensorInfo()
150 if (inputs_tensor_map_.size() != inputs_shape_.size()) { in InferMirrorOps()
156 if (CreateGroupByTensorMap(inputs_tensor_map_[0], &group) != SUCCESS) { in InferMirrorOps()
Dbatchnorm_info.cc130 inputs_tensor_map_.push_back(input_tensor_map); // input in InferTensorMap()
131 inputs_tensor_map_.push_back(in_other_tensor_map); // scale in InferTensorMap()
132 inputs_tensor_map_.push_back(in_other_tensor_map); // bias in InferTensorMap()
133 inputs_tensor_map_.push_back(in_other_tensor_map); // mean in InferTensorMap()
134 inputs_tensor_map_.push_back(in_other_tensor_map); // variance in InferTensorMap()
136 outputs_tensor_map_ = inputs_tensor_map_; in InferTensorMap()
Dtopk_info.cc78 inputs_tensor_map_.push_back(tensor_map); in InferTensorMap()
114 if (inputs_tensor_map_.size() != inputs_shape_.size()) { in InferMirrorOps()
120 for (size_t i = 0; i < inputs_tensor_map_.size(); ++i) { in InferMirrorOps()
122 if (CreateGroupByTensorMap(inputs_tensor_map_[i], &group) != SUCCESS) { in InferMirrorOps()
Dgather_v2_info.cc155 inputs_tensor_map_.emplace_back(std::move(tensor_map_in)); in InferTensorMap()
156 inputs_tensor_map_.emplace_back(std::move(tensor_map_in_index)); in InferTensorMap()
172 if (inputs_tensor_map_.size() != GATHER_V2_INPUTS_SIZE) { in InferTensorInfo()
174 << inputs_tensor_map_.size(); in InferTensorInfo()
188 …if ((input_tensor_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_.at(0), input_shape) … in InferTensorInfo()
189 …(input_index_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_.at(1), input_index_shape)… in InferTensorInfo()
Dscatter_update_info.cc126 inputs_tensor_map_.push_back(input_tensor_map); // input in InferTensorMap()
127 inputs_tensor_map_.push_back(indices_tensor_map); // indices in InferTensorMap()
128 inputs_tensor_map_.push_back(updates_tensor_map); // updates in InferTensorMap()
Dactivation_info.cc219 Shape tensor_map = inputs_tensor_map_[0]; in InferMirrorOps()
253 inputs_tensor_map_.push_back(tensor_map_index); in InferTensorMap()
290 inputs_tensor_map_.push_back(tensor_map_in); in InferTensorMap()
359 Shape tensor_map = inputs_tensor_map_[0]; in InferMirrorOps()
429 inputs_tensor_map_.push_back(input_tensor_map); in InferTensorMap()
469 if (inputs_tensor_map_.empty()) { in InferMirrorOps()
475 if (CreateGroupByTensorMap(inputs_tensor_map_[0], &group) != SUCCESS) { in InferMirrorOps()
568 inputs_tensor_map_.push_back(input_tensor_map); in InferTensorMap()
Ddsd_matmul_info.cc144 inputs_tensor_map_.push_back(input_tensor_map_w1); in InferTensorMap()
145 inputs_tensor_map_.push_back(input_tensor_map_w2); in InferTensorMap()
146 inputs_tensor_map_.push_back(input_tensor_map_v); in InferTensorMap()
Dmatmul_dds_info.cc192 inputs_tensor_map_.push_back(input_tensor_map_q); in InferTensorMap()
193 inputs_tensor_map_.push_back(input_tensor_map_k); in InferTensorMap()
194 inputs_tensor_map_.push_back(input_tensor_map_local_mask); in InferTensorMap()
195 inputs_tensor_map_.push_back(input_tensor_map_global_mask); in InferTensorMap()
Dprelu_info.cc80 inputs_tensor_map_.push_back(input_tensor_map); in InferTensorMap()
81 inputs_tensor_map_.push_back(param_tensor_map); in InferTensorMap()
Dbias_add_info.cc69 inputs_tensor_map_.push_back(sub_a_tensor_map); in InferTensorMap()
70 inputs_tensor_map_.push_back(sub_b_tensor_map); in InferTensorMap()
Dstrided_slice_info.cc172 inputs_tensor_map_.push_back(tensor_map); in InferTensorMap()
179 if (inputs_tensor_map_.empty()) { in InferMirrorOps()
183 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps()
Dslice_info.cc120 inputs_tensor_map_.push_back(tensor_map); in InferTensorMap()
127 if (inputs_tensor_map_.empty()) { in InferMirrorOps()
131 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps()
Dgather_v2_p_info.cc383 Shape input_a_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps()
486 inputs_tensor_map_.emplace_back(std::move(tensor_map_params)); in InferInputsTensorMap()
487 inputs_tensor_map_.emplace_back(std::move(tensor_map_index)); in InferInputsTensorMap()
542 (void)inputs_tensor_map_.emplace_back(std::move(param_map)); in InferTensorMap()
543 (void)inputs_tensor_map_.emplace_back(std::move(indices_map)); in InferTensorMap()
552 (void)inputs_tensor_map_.emplace_back(std::move(param_tensor_map)); // param in InferTensorMap()
553 (void)inputs_tensor_map_.emplace_back(std::move(indices_tensor_map)); // indices in InferTensorMap()
575 …if ((input_tensor_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_.at(0), input_shape) … in InferTensorInfo()
576 …(input_index_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_.at(1), input_index_shape)… in InferTensorInfo()
Dgathernd_info.cc108 inputs_tensor_map_.push_back(input_tensor_map); in InferTensorMap()
109 inputs_tensor_map_.push_back(indices_tensor_map); in InferTensorMap()
Dloss_info.cc88 inputs_tensor_map_.push_back(tensor_map_index); // input in InferTensorMap()
89 inputs_tensor_map_.push_back(tensor_map_index); // label in InferTensorMap()
Dreduce_method_info.cc130 inputs_tensor_map_.push_back(tensor_map_index); in InferTensorMap()
285 Shape input_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps()
310 Shape input_tensor_map = inputs_tensor_map_.at(0); in InferMirrorOps()
365 …if ((input_tensor_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[0], input_shape) != … in InferTensorInfo()
494 …if ((input_tensor_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[0], input_shape) != … in InferTensorInfo()
Dtensordot_info.cc228 inputs_tensor_map_.push_back(input_b_tensor_map); in InferTensorMapAxesInt()
273 inputs_tensor_map_.push_back(input_b_tensor_map); in InferTensorMapAxesTuple()
295 inputs_tensor_map_.push_back(input_a_tensor_map); in InferTensorMap()
Dunsorted_segment_op_info.cc104 Shape tensor_map = inputs_tensor_map_[0]; in InferMirrorOps()
160 inputs_tensor_map_.emplace_back(std::move(tensor_map_in)); in InferTensorMap()
161 inputs_tensor_map_.emplace_back(std::move(tensor_map_in_index)); in InferTensorMap()
Dtile_info.cc132 inputs_tensor_map_.push_back(input_tensor_map); in InferTensorMap()
139 Shape input_tensor_map = inputs_tensor_map_[0]; in InferMirrorOps()
Dconv2d_info.cc492 (void)inputs_tensor_map_.emplace_back(std::move(input_tensor_map)); in InferTensorMap()
493 (void)inputs_tensor_map_.emplace_back(std::move(weight_tensor_map)); in InferTensorMap()
1001 (void)inputs_tensor_map_.emplace_back(std::move(input_tensor_map)); in InferTensorMap()
1002 (void)inputs_tensor_map_.emplace_back(std::move(weight_tensor_map)); in InferTensorMap()
1014 if (inputs_tensor_map_.size() != inputs_shape_.size()) { in InferMirrorOps()
1020 for (size_t i = 0; i < inputs_tensor_map_.size(); ++i) { in InferMirrorOps()
1022 if (CreateGroupByTensorMap(inputs_tensor_map_[i], &group) != SUCCESS) { in InferMirrorOps()
Darithmetic_info.cc172 inputs_tensor_map_.push_back(SetTensorMap(sub_a_expend_strategy, dev_shape, sub_a_strategy)); in InferTensorMap()
173 inputs_tensor_map_.push_back(SetTensorMap(sub_b_expend_strategy, dev_shape, sub_b_strategy)); in InferTensorMap()
Dmatmul_info.cc308 inputs_tensor_map_.push_back(mat_a_tensor_map); in InferTensorMap()
309 inputs_tensor_map_.push_back(mat_b_tensor_map); in InferTensorMap()
327 …if ((mat_a_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[0], inputs_shape_[0]) != SU… in InferTensorLayout()
328 …(mat_b_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[1], inputs_shape_[1]) != SUCCES… in InferTensorLayout()
Dtmp_identity_info.cc46 inputs_tensor_map_.push_back(tensor_map_index); in InferTensorMap()
Dlayer_norm_info.cc133 inputs_tensor_map_.push_back(tensor_map); in CreateInputTensorMap()
145 Shape first_output_tensor_map = inputs_tensor_map_[0]; in InferTensorMap()
Dreshape_info.cc168 inputs_tensor_map_.push_back(tensor_map_index_input); in InferTensorMap()
204 TensorMap tensor_map_array_in = inputs_tensor_map_.at(0); in InferTensorLayout()
344 inputs_tensor_map_.push_back(input_layout_.tensor_map().array()); in Init()

12