Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dreshape_info.cc250 outputs_tensor_info_.push_back(tensor_info_in); in InferTensorInfo()
275 outputs_tensor_info_.push_back(tensor_info_out); in InferTensorInfo()
283 outputs_tensor_info_.push_back(tensor_info_out); in InferTensorInfoByLayout()
397 operator_cost()->GetForwardComputationCost(inputs_tensor_info_, outputs_tensor_info_, stage_id); in SetCostForReshape()
398 …unication_cost = operator_cost()->GetCommCost(inputs_tensor_info_, outputs_tensor_info_, stage_id); 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()
Dget_next_info.cc78 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()
207 …(void)std::transform(outputs_tensor_info_.begin(), outputs_tensor_info_.end(), std::back_inserter(… in InferReplaceOps()
Dmatmul_info.cc359 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()
615 …operator_cost()->GetForwardComputationCost(relica_inputs_tensor_vector, outputs_tensor_info_, stag… in SetCostUnderStrategy()
616 …n_cost = operator_cost()->GetCommCost(relica_inputs_tensor_vector, outputs_tensor_info_, stage_id); in SetCostUnderStrategy()
620 … operator_cost()->GetForwardCommCost(relica_inputs_tensor_vector, outputs_tensor_info_, stage_id); in SetCostUnderStrategy()
635 std::make_shared<StrategyWithCost>(strategy, inputs_tensor_info_, outputs_tensor_info_); in SetCostUnderStrategy()
Dreduce_method_info.cc376 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()
505 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()
506 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()
Doperator_info.h123 std::vector<TensorInfo> outputs_tensor_info() const { return outputs_tensor_info_; } in outputs_tensor_info()
237 std::vector<TensorInfo> outputs_tensor_info_; variable
Doperator_info.cc138 outputs_tensor_info_.clear(); in ResetQueueMember()
225 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()
1381 operator_cost()->GetForwardComputationCost(inputs_tensor_info_, outputs_tensor_info_, stage_id); in SetCostUnderStrategyBase()
1382 …unication_cost = operator_cost()->GetCommCost(inputs_tensor_info_, outputs_tensor_info_, stage_id); 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()
Dbroadcast_to_info.cc162 Shape to_shape = outputs_tensor_info_[0].slice_shape(); in ComputeReplaceGraph()
Dgatherd_info.cc138 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()
Dgather_v2_info.cc200 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()
Dgather_v2_p_info.cc592 outputs_tensor_info_.push_back(output_tensor_info); in InferTensorInfo()