Searched refs:OutPutInfoVector (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_redistribution.cc | 52 OutPutInfoVector output_info_vector; in InferTensorRedistributionOperatorListUnExpand()
74 return std::make_shared<std::pair<OperatorVector, OutPutInfoVector>>( in InferTensorRedistributionOperatorListUnExpand()
111 OutPutInfoVector output_info_vector; in InferTensorRedistributionOperatorList()
121 return std::make_shared<std::pair<OperatorVector, OutPutInfoVector>>( in InferTensorRedistributionOperatorList()
127 OutPutInfoVector *const output_info_vector) { in InferReshape()
177 … OutPutInfoVector *const output_info_vector, bool is_cost_model) { in InferRedistribution()
|
| 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 | redistribution_operator_infer.h | 49 OutPutInfoVector output_info_vector() const { return output_info_vector_; } in output_info_vector()
63 OutPutInfoVector output_info_vector_;
|
| /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/auto_parallel/ |
| D | costmodel.h | 40 using OutPutInfoVector = std::vector<OutPutInfo>; variable
44 using RedistributionOpListPtr = std::shared_ptr<std::pair<OperatorVector, OutPutInfoVector>>;
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.h | 116 OutPutInfoVector replace_op_info() const { return replace_op_info_; } in replace_op_info()
247 OutPutInfoVector replace_op_info_;
|