Home
last modified time | relevance | path

Searched refs:GetInputNumber (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/
Dstrategy_test.cc33 TEST_F(TestStrategy, GetInputNumber) { in TEST_F() argument
41 int32_t number_test = strategy.GetInputNumber(); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstrategy.h54 size_t GetInputNumber() const { return inputs_.size(); } in GetInputNumber() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dbatch_parallel_info.cc35 size_t strategy_size = strategy->GetInputNumber(); in CheckStrategy()
Doperator_info.cc65 size_t strategy_size = strategy->GetInputNumber(); in CheckStrategyValue()
1044 for (size_t i = 0; i < strategy->GetInputNumber(); ++i) { in PrintStrategy()
1053 if (i != strategy->GetInputNumber() - 1) { in PrintStrategy()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_generate_strategy.cc740 if (strategy->GetInputNumber() == 0) { in PrepareIncomingOperatorInputStrategy()
1195 ops[i]->selected_strategy()->GetInputNumber() != 0) { in CopyOutgoingOperatorInputStrategy()