Searched defs:OperatorInfo (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | operator_info.h | 45 class OperatorInfo; variable
49 OperatorInfo(const std::string &name, const SplitStrategy &strategy) in OperatorInfo() function
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | tmp_identity_info.h | 37 … : OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<TmpIdentityCost>()) {} in OperatorInfo() function
|
| D | gather_v2_p_info.h | 36 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<GatherV2PCost>()), in OperatorInfo() function
|
| D | operator_info.h | 57 …OperatorInfo(std::string name, Shapes inputs_shape, Shapes outputs_shape, PrimitiveAttrs attrs, Op… in OperatorInfo() function
|