Home
last modified time | relevance | path

Searched refs:StrategyWithCost (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dcostmodel.h79 class StrategyWithCost {
81 …StrategyWithCost(StrategyPtr strategy, std::vector<TensorInfo> inputs_, std::vector<TensorInfo> ou… in StrategyWithCost() function
83 StrategyWithCost(StrategyPtr strategy, CostPtrList c_list) in StrategyWithCost() function
86 StrategyWithCost(const StrategyWithCost &swc) = delete;
87 StrategyWithCost(StrategyWithCost &&swc) in StrategyWithCost() function
92 ~StrategyWithCost() = default;
Dgraph_costmodel.cc929 std::vector<std::shared_ptr<StrategyWithCost>> op1_new_stra_cost; in EliminationSources()
944 …std::shared_ptr<StrategyWithCost> swc = std::make_shared<StrategyWithCost>(op1_new_stra, op1_new_c… in EliminationSources()
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dtmpidentity_test.cc140 std::vector<std::shared_ptr<StrategyWithCost>> sc = identity_ptr->GetStrategyCost(); in TEST_F()
157 std::vector<std::shared_ptr<StrategyWithCost>> sc = identity_ptr->GetStrategyCost(); in TEST_F()
168 std::vector<std::shared_ptr<StrategyWithCost>> sc = identity_ptr2->GetStrategyCost(); in TEST_F()
Dactivation_test.cc76 std::vector<std::shared_ptr<StrategyWithCost>> sc = act_ptr_->GetStrategyCost(); in TEST_F()
96 std::vector<std::shared_ptr<StrategyWithCost>> sc = soft_ptr_->GetStrategyCost(); in TEST_F()
Dtensor_add_info_test.cc184 std::vector<std::shared_ptr<StrategyWithCost>> sc = tensor_add->GetStrategyCost(); in TEST_F()
206 std::vector<std::shared_ptr<StrategyWithCost>> sc = tensor_add1->GetStrategyCost(); in TEST_F()
Dprelu_test.cc157 std::vector<std::shared_ptr<StrategyWithCost>> sc = prelu->GetStrategyCost(); in TEST_F()
262 std::vector<std::shared_ptr<StrategyWithCost>> sc = prelu_2d->GetStrategyCost(); in TEST_F()
Dtranspose_test.cc183 std::vector<std::shared_ptr<StrategyWithCost>> sc = transpose->GetStrategyCost(); in TEST_F()
Dmatmul_info_test.cc565 std::vector<std::shared_ptr<StrategyWithCost>> sc = matmul1->GetStrategyCost(); in TEST_F()
581 std::vector<std::shared_ptr<StrategyWithCost>> sc = matmul3->GetStrategyCost(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dreshape_info.h61 Status GenetateStrategyCosts(const std::vector<std::shared_ptr<StrategyWithCost>> &pre_stra_costs,
62 … const std::vector<std::shared_ptr<StrategyWithCost>> &next_stra_costs, int64_t out_index,
Doperator_info.h103 std::vector<std::shared_ptr<StrategyWithCost>> GetStrategyCost() { return strategy_cost_; } in GetStrategyCost()
104 void SetStrategyCost(const std::vector<std::shared_ptr<StrategyWithCost>> &);
124 std::vector<std::shared_ptr<StrategyWithCost>> strategy_cost() const { return strategy_cost_; } in strategy_cost()
259 std::vector<std::shared_ptr<StrategyWithCost>> strategy_cost_;
Dreshape_info.cc411 std::shared_ptr<StrategyWithCost> swc = in SetCostForReshape()
412 std::make_shared<StrategyWithCost>(strategy, inputs_tensor_info_, outputs_tensor_info_); in SetCostForReshape()
444 Status ReshapeInfo::GenetateStrategyCosts(const std::vector<std::shared_ptr<StrategyWithCost>> &pre… in GenetateStrategyCosts()
445 … const std::vector<std::shared_ptr<StrategyWithCost>> &next_stra_costs, in GenetateStrategyCosts()
Doperator_info.cc1396 std::shared_ptr<StrategyWithCost> swc = in SetCostUnderStrategyBase()
1397 std::make_shared<StrategyWithCost>(strategy, inputs_tensor_info_, outputs_tensor_info_); in SetCostUnderStrategyBase()
1418 std::vector<std::shared_ptr<StrategyWithCost>> ret; in ApproximateStrategies()
1425 …[&alpha, &beta](const std::shared_ptr<StrategyWithCost> &s1, const std::shared_ptr<StrategyWithCos… in ApproximateStrategies()
1743 void OperatorInfo::SetStrategyCost(const std::vector<std::shared_ptr<StrategyWithCost>> &stra_cost)… in SetStrategyCost()
Dmatmul_info.cc634 std::shared_ptr<StrategyWithCost> swc = in SetCostUnderStrategy()
635 std::make_shared<StrategyWithCost>(strategy, inputs_tensor_info_, outputs_tensor_info_); in SetCostUnderStrategy()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc866 std::vector<std::shared_ptr<StrategyWithCost>> pre_stra_costs; in ReshapeCostCompute()
883 std::vector<std::shared_ptr<StrategyWithCost>> next_stra_costs; in ReshapeCostCompute()