Home
last modified time | relevance | path

Searched defs:stra (Results 1 – 25 of 55) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Duniform_real_info.cc56 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
66 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
79 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferTensorMap() local
128 std::vector<Dimensions> stra = strategy_->GetInputDim(); in UpdateShape() local
145 std::vector<Dimensions> stra = strategy_->GetInputDim(); in ReplaceNodeInputOrAttrs() local
Dbias_add_info.cc34 Strategys stra = strategy->GetInputDim(); in CheckStrategy() local
47 Strategys stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
61 Strategys stra = strategy_->GetInputDim(); in InferTensorMap() local
Dvirtual_dataset_info.cc39 Strategys stra = strategy->GetInputDim(); in CheckStrategy() local
87 Strategys stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
103 auto stra = strategy_->GetInputDim(); in InferTensorMap() local
Dresizebilinear_info.cc58 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
82 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dprelu_info.cc41 Strategys stra = strategy->GetInputDim(); in CheckStrategy() local
57 Strategys stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dmaxpool_info.cc133 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
158 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dreluv2_info.cc41 Strategys stra = strategy->GetInputDim(); in CheckStrategy() local
68 Strategys stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dselect_info.cc48 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
64 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dgathernd_info.cc41 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
69 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dtopk_info.cc38 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
54 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dreduce_method_info.cc35 Strategys stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
147 Dimensions stra = strategy_->GetInputDim().at(0); in InferForwardCommunication() local
227 Dimensions stra = strategy_->GetInputDim().at(0); in InferForwardCommunication() local
333 Dimensions stra = strategy_->GetInputDim().at(0); in InferOutputStrategy() local
451 Strategys stra = strategy->GetInputDim(); in CheckStrategy() local
549 Dimensions stra = strategy->GetInputDim().at(0); in CheckStrategy() local
Dscatter_update_info.cc47 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
96 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dsplit_info.cc84 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
105 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dloss_info.cc36 Strategys stra = strategy->GetInputDim(); in CheckStrategy() local
73 Strategys stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dget_next_info.cc115 for (Dimensions stra : stras) { in CheckStrategy() local
134 for (auto &stra : dataset_strategy_) { in CheckStrategy() local
232 Strategys stra; in GenerateOpStrategies() local
Dpack_info.cc67 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
88 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dconcat_info.cc68 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
110 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dbatchnorm_info.cc72 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
100 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dgatherd_info.cc66 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
89 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dbroadcast_to_info.cc68 auto stra = strategy->GetInputDim().at(0); in CheckStrategy() local
81 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dstrided_slice_info.cc106 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
149 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dlayer_norm_info.cc59 Strategys stra = strategy->GetInputDim(); in CheckStrategy() local
114 Strategys stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Dvirtual_output_info.cc37 Strategys stra = strategy->GetInputDim(); in CheckStrategy() local
Duniform_candidate_sampler_info.cc86 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy() local
105 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape() local
Donehot_info.cc56 auto stra = strategy->GetInputDim().at(0); in CheckStrategy() local
72 Strategys stra = strategy_->GetInputDim(); in InferDevMatrixShape() local

123