Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dgenerate_strategy_test.cc58 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
71 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
86 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
101 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
115 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
138 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
154 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dloss_info.cc146 …if (GenerateStrategiesWithBroadcast(stage_id, inputs_shape_, splittable_inputs, &sp_vector) != SUC… in GenerateOpStrategies()
Darithmetic_info.cc187 …if (GenerateStrategiesWithBroadcast(stage_id, inputs_shape_, splittable_inputs, &sp_vector) != SUC… in GenerateOpStrategies()
Doperator_info.h325 Status GenerateStrategiesWithBroadcast(int64_t stage_id, const Shapes &inputs_shape, const Shapes &…
Doperator_info.cc1319 Status GenerateStrategiesWithBroadcast(int64_t stage_id, const Shapes &inputs_shape, const Shapes &… in GenerateStrategiesWithBroadcast() function