| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_redistribution.cc | 51 OperatorVector operator_vector; in InferTensorRedistributionOperatorListUnExpand()
74 return std::make_shared<std::pair<OperatorVector, OutPutInfoVector>>( in InferTensorRedistributionOperatorListUnExpand()
110 OperatorVector operator_vector; in InferTensorRedistributionOperatorList()
121 return std::make_shared<std::pair<OperatorVector, OutPutInfoVector>>( in InferTensorRedistributionOperatorList()
126 OperatorVector *const operator_vector, in InferReshape()
176 OperatorVector *const operator_vector, in InferRedistribution()
|
| D | redistribution_operator_infer.h | 48 OperatorVector operator_vector() const { return operator_vector_; } in operator_vector()
62 OperatorVector operator_vector_;
|
| D | tensor_redistribution.h | 63 … OperatorVector *const operator_vector, OutPutInfoVector *const output_info_vector);
65 … OperatorVector *const operator_vector, OutPutInfoVector *const output_info_vector,
|
| D | construct_operator.h | 38 OperatorVector SkipRedisReshapeOP(const Shape &shape);
|
| D | construct_operator.cc | 33 OperatorVector ConstructOperator::SkipRedisReshapeOP(const Shape &shape) { in SkipRedisReshapeOP()
40 OperatorVector opvector; in SkipRedisReshapeOP()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.h | 43 using ForwardOp = OperatorVector;
44 using MirrorOps = std::vector<OperatorVector>;
45 using Ops = std::vector<OperatorVector>;
46 using VirtualDivOp = OperatorVector;
311 OperatorVector CreateMirrorOps(const std::string &group_name, size_t dev_num);
|
| D | topk_info.cc | 128 OperatorVector mirror_op; in InferMirrorOps()
146 OperatorVector tmp_mirror_op; // tmp mirror op for 'k' in InferMirrorOps()
|
| D | gather_v2_info.cc | 204 OperatorVector CreateSubOp(int64_t sub_value) { in CreateSubOp()
205 OperatorVector ops; in CreateSubOp()
248 OperatorVector sub_op; in InferTensorSubOps()
|
| D | unsorted_segment_op_info.cc | 111 OperatorVector mirror_op; in InferMirrorOps()
112 OperatorVector op_for_value; in InferMirrorOps()
113 OperatorVector op_for_value2; in InferMirrorOps()
|
| D | gatherd_info.cc | 166 OperatorVector mirror_op = CreateMirrorOps(group[0].name(), group[0].GetDevNum()); in InferMirrorOps()
169 OperatorVector tmp_mirror_op; // dim in InferMirrorOps()
|
| D | activation_info.cc | 226 OperatorVector mirror_op; in InferMirrorOps()
366 OperatorVector mirror_op; in InferMirrorOps()
367 OperatorVector op_for_value; in InferMirrorOps()
485 OperatorVector mirror_op, placeholder_op; in InferMirrorOps()
|
| D | reduce_method_info.cc | 292 OperatorVector op_for_weight; in InferMirrorOps()
293 OperatorVector op_for_reduce_axis; // helper node in InferMirrorOps()
317 OperatorVector op_for_weight; in InferMirrorOps()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | matmul_info_test.cc | 351 OperatorVector forward_op = matmul1->forward_op(); in TEST_F()
377 OperatorVector forward_op = matmul1->forward_op(); in TEST_F()
387 OperatorVector virtual_div_op = matmul1->virtual_div_op(); in TEST_F()
407 OperatorVector mirror_op = mirror_ops.at(1); in TEST_F()
427 OperatorVector mirror_op = mirror_ops.at(1); in TEST_F()
447 OperatorVector mirror_op = mirror_ops.at(1); in TEST_F()
478 OperatorVector mirror_op = mirror_ops.at(1); in TEST_F()
|
| D | reduce_method_test.cc | 128 OperatorVector forward_op = reduce_sum->forward_op(); in TEST_F()
139 OperatorVector forward_op = reduce_sum->forward_op(); in TEST_F()
176 OperatorVector mirror_op = mirror_ops.at(0); in TEST_F()
|
| D | activation_info_test.cc | 124 OperatorVector forward_op = activation->forward_op(); in TEST_F()
137 OperatorVector mirror_op = mirror_ops.at(0); in TEST_F()
|
| D | l2_normalize_info_test.cc | 124 OperatorVector forward_op = norm->forward_op(); in TEST_F()
180 OperatorVector mirror_op = mirror_ops.at(0); in TEST_F()
|
| D | softmax_entropy_loss_info_test.cc | 140 OperatorVector forward_op = loss->forward_op(); in TEST_F()
163 OperatorVector virtual_div_op = loss->virtual_div_op(); in TEST_F()
|
| D | tensor_add_info_test.cc | 132 OperatorVector forward_op = tensor_add->forward_op(); in TEST_F()
232 OperatorVector mirror_op = mirror_ops.at(1); in TEST_F()
|
| D | prelu_test.cc | 129 OperatorVector mirror_op = mirror_ops.at(1); in TEST_F()
234 OperatorVector mirror_op = mirror_ops.at(1); in TEST_F()
|
| D | tanh_info_test.cc | 123 OperatorVector forward_op = tanh->forward_op(); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | tensor_redistribution_test.cc | 89 std::shared_ptr<std::pair<OperatorVector, OutPutInfoVector>> op_ptr; in TEST_F()
142 std::shared_ptr<std::pair<OperatorVector, OutPutInfoVector>> op_ptr; in TEST_F()
195 std::shared_ptr<std::pair<OperatorVector, OutPutInfoVector>> op_ptr; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 58 void ForwardCommunication(OperatorVector forward_op, const CNodePtr &node);
81 void StepReplaceOp(OperatorVector replace_op, const CNodePtr &node);
|
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | virtual_dataset_test.cc | 79 OperatorVector forward_op = virtual_dataset->forward_op(); in TEST_F()
|
| D | step_parallel_test.cc | 403 OperatorVector op_list = {op, op}; in TEST_F()
442 OperatorVector op_list; in TEST_F()
465 OperatorVector op_list; in TEST_F()
484 OperatorVector op_list = {op}; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | costmodel.h | 43 using OperatorVector = std::vector<Operator>; variable
44 using RedistributionOpListPtr = std::shared_ptr<std::pair<OperatorVector, OutPutInfoVector>>;
|