Home
last modified time | relevance | path

Searched refs:inputs_tensor_info_ (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dmatmul_info.cc357 inputs_tensor_info_.push_back(mat_a_tensor_info); in InferTensorInfo()
358 inputs_tensor_info_.push_back(mat_b_tensor_info); in InferTensorInfo()
539 Shape replica_input0_shape(inputs_tensor_info_[0].shape()); in InitTensorInfoForCost()
540 Shape replica_input0_slice_shape(inputs_tensor_info_[0].slice_shape()); in InitTensorInfoForCost()
551 relica_inputs_tensor_vector->push_back(inputs_tensor_info_[0]); in InitTensorInfoForCost()
554 Shape replica_input1_shape(inputs_tensor_info_[1].shape()); in InitTensorInfoForCost()
555 Shape replica_input1_slice_shape(inputs_tensor_info_[1].slice_shape()); in InitTensorInfoForCost()
566 relica_inputs_tensor_vector->push_back(inputs_tensor_info_[1]); in InitTensorInfoForCost()
576 if (inputs_tensor_info_.empty()) { in CheckForTensorSliceValid()
579 for (auto &one_input_tensor : inputs_tensor_info_) { in CheckForTensorSliceValid()
[all …]
Dreshape_info.cc249 inputs_tensor_info_.push_back(tensor_info_in); in InferTensorInfo()
274 inputs_tensor_info_.push_back(tensor_info_in); in InferTensorInfo()
282 inputs_tensor_info_.push_back(tensor_info_in); in InferTensorInfoByLayout()
397 operator_cost()->GetForwardComputationCost(inputs_tensor_info_, outputs_tensor_info_, stage_id); in SetCostForReshape()
398 …double communication_cost = operator_cost()->GetCommCost(inputs_tensor_info_, outputs_tensor_info_… in SetCostForReshape()
402 operator_cost()->GetForwardCommCost(inputs_tensor_info_, outputs_tensor_info_, stage_id); in SetCostForReshape()
412 std::make_shared<StrategyWithCost>(strategy, inputs_tensor_info_, outputs_tensor_info_); in SetCostForReshape()
Dgatherd_info.cc125 inputs_tensor_info_.push_back(input_tensor_info); in InferTensorInfo()
129 (void)inputs_tensor_info_.insert(inputs_tensor_info_.begin() + 1, dim_tensor_info); in InferTensorInfo()
Ddropout_do_mask_info.cc225 if (inputs_tensor_info_.empty()) { in GetDropoutGenMaskReplaceOp()
244 Shape input_slice_shape = inputs_tensor_info_[0].slice_shape(); in GetDropoutGenMaskReplaceOp()
Dgather_v2_info.cc198 inputs_tensor_info_.push_back(input_tensor_info); in InferTensorInfo()
199 inputs_tensor_info_.push_back(input_index_info); in InferTensorInfo()
Doperator_info.h122 std::vector<TensorInfo> inputs_tensor_info() const { return inputs_tensor_info_; } in inputs_tensor_info()
236 std::vector<TensorInfo> inputs_tensor_info_; variable
Doperator_info.cc137 inputs_tensor_info_.clear(); in ResetQueueMember()
215 inputs_tensor_info_.push_back(input_tensor_info); in InferTensorInfo()
1381 operator_cost()->GetForwardComputationCost(inputs_tensor_info_, outputs_tensor_info_, stage_id); in SetCostUnderStrategyBase()
1382 …double communication_cost = operator_cost()->GetCommCost(inputs_tensor_info_, outputs_tensor_info_… in SetCostUnderStrategyBase()
1386 operator_cost()->GetForwardCommCost(inputs_tensor_info_, outputs_tensor_info_, stage_id); in SetCostUnderStrategyBase()
1397 std::make_shared<StrategyWithCost>(strategy, inputs_tensor_info_, outputs_tensor_info_); in SetCostUnderStrategyBase()
1730 return operator_cost()->GetForwardComputationCost(inputs_tensor_info_, outputs_tensor_info_, 0); in GetForwardMemoryCostFromCNode()
Dslice_info.cc221 Shape input_slice_shape = inputs_tensor_info_[0].slice_shape(); in ComputeReplaceGraph()
Dreduce_method_info.cc375 inputs_tensor_info_.push_back(input_tensor_info); in InferTensorInfo()
504 inputs_tensor_info_.push_back(input_tensor_info); in InferTensorInfo()
Dgather_v2_p_info.cc590 inputs_tensor_info_.push_back(input_tensor_info); in InferTensorInfo()
591 inputs_tensor_info_.push_back(input_index_info); in InferTensorInfo()