Searched refs:GetInputStage (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | strategy.h | 43 Strategy(const Strategy &another_stra) : stage_(another_stra.GetInputStage()) { in Strategy()
56 int64_t GetInputStage() const { return stage_; } in GetInputStage() function
71 if ((stage_ != another_stra->GetInputStage()) || (inputs_ != another_stra->GetInputDim())) { in IsEqual()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | activation_test.cc | 87 …E_EQ(act_ptr_->operator_cost()->GetComputationCost(inputs_info, outputs_info, sp->GetInputStage()), in TEST_F()
89 …T_DOUBLE_EQ(act_ptr_->operator_cost()->GetCommCost(inputs_info, outputs_info, sp->GetInputStage()), in TEST_F()
112 …_EQ(soft_ptr_->operator_cost()->GetComputationCost(inputs_info, outputs_info, sp->GetInputStage()), in TEST_F()
114 …_DOUBLE_EQ(soft_ptr_->operator_cost()->GetCommCost(inputs_info, outputs_info, sp->GetInputStage()), in TEST_F()
|
| D | tensor_add_info_test.cc | 191 … = tensor_add->operator_cost()->GetComputationCost(inputs_info, outputs_info, sp->GetInputStage()); in TEST_F()
195 …m_cost0 = tensor_add->operator_cost()->GetCommCost(inputs_info, outputs_info, sp->GetInputStage()); in TEST_F()
213 …= tensor_add1->operator_cost()->GetComputationCost(inputs_info, outputs_info, sp->GetInputStage()); in TEST_F()
217 …_cost0 = tensor_add1->operator_cost()->GetCommCost(inputs_info, outputs_info, sp->GetInputStage()); in TEST_F()
|
| D | tmpidentity_test.cc | 148 …(identity_ptr->operator_cost()->GetComputationCost(inputs_info, outputs_info, sp->GetInputStage()), in TEST_F()
150 …UBLE_EQ(identity_ptr->operator_cost()->GetCommCost(inputs_info, outputs_info, sp->GetInputStage()), in TEST_F()
|
| D | matmul_info_test.cc | 572 …LE_EQ(matmul1->operator_cost()->GetComputationCost(inputs_info, outputs_info, sp->GetInputStage()), in TEST_F()
602 …tmul3->operator_cost()->GetComputationCost(replica_inputs_info, outputs_info, sp->GetInputStage()), in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | strategy_test.cc | 45 TEST_F(TestStrategy, GetInputStage) { in TEST_F() argument
52 int32_t stage_test = strategy.GetInputStage(); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reshape_info.cc | 395 int64_t stage_id = strategy->GetInputStage(); in SetCostForReshape()
469 …eshape_stra = std::make_shared<Strategy>(pre_stra_cost->strategy_ptr->GetInputStage(), stra_inputs… in GenetateStrategyCosts()
|
| D | matmul_info.cc | 611 int64_t stage_id = strategy->GetInputStage(); in SetCostUnderStrategy()
|
| D | operator_info.cc | 1379 int64_t stage_id = strategy->GetInputStage(); in SetCostUnderStrategyBase()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/ |
| D | parallel_strategy_checkpoint.cc | 153 parallel_strategys->set_stage(UlongToUint(LongToUlong(node_stra.second->GetInputStage()))); in Save()
|
| /third_party/mindspore/mindspore/ccsrc/debug/ |
| D | draw.cc | 456 …std::vector<ValuePtr> temp = {MakeValue(strategyPtr->GetInputStage()), MakeValue(strategyPtr->GetI… in DrawParallelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | graph_costmodel.cc | 1608 …std::make_shared<Strategy>((*pre_iter)->prev_operator()->strategy()->GetInputStage(), stra_inputs); in InitReshapeStrategy()
|