Home
last modified time | relevance | path

Searched defs:input_strategy (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Duniform_candidate_sampler_info.cc92 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
179 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateBatchStrategies() local
204 auto input_strategy = strategy_->GetInputDim().at(0); in replace_graph() local
216 auto input_strategy = strategy_->GetInputDim().at(0); in ComputeReplaceGraph() local
Dreluv2_info.cc42 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
69 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
Dloss_info.cc37 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
74 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
Dlayer_norm_info.cc70 Dimensions input_strategy = stra[LAYER_NORM_INPUT_INDEX]; in CheckStrategy() local
186 Dimensions input_strategy = sp->GetInputDim()[0]; in GenerateGammaAndBetaStrategies() local
Dl2_normalize_info.cc35 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
Dtmp_identity_info.cc33 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
Dresizebilinear_info.cc64 Dimensions input_strategy = stra[0]; in CheckStrategy() local
Dprelu_info.cc58 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
Dmaxpool_info.cc139 Dimensions input_strategy = stra[0]; in CheckStrategy() local
Dgathernd_info.cc144 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateOpStrategies() local
Dsplit_info.cc163 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateBatchStrategies() local
Ddsd_matmul_info.cc94 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
Dgatherd_info.cc72 Dimensions input_strategy = stra[0]; in CheckStrategy() local
Dactivation_info.cc111 Dimensions input_strategy = stra.at(0); in CheckStrategy() local
209 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
Dconv2d_info.cc266 Dimensions input_strategy = stra[0]; in CheckStrategyBase() local
316 Dimensions input_strategy = stra[0]; in CheckStrategy() local
911 Dimensions input_strategy = stra[0]; in CheckStrategy() local
Donehot_info.cc73 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
Dslice_info.cc197 auto input_strategy = strategy_->GetInputDim().at(0); in replace_graph() local
Dmatmul_dds_info.cc114 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
Dreduce_method_info.cc36 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape() local
453 Shape input_strategy = stra.at(0); in CheckStrategy() local
Dget_next_info.cc128 Dimensions input_strategy; in CheckStrategy() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dpipeline_split_utils.cc101 Dimensions input_strategy; in SetStridedSliceStrategy() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc1771 Dimensions input_strategy; in SetVirtualDatasetStrategy() local