Home
last modified time | relevance | path

Searched refs:operator_attrs (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dreduce_method_test.cc141 OperatorAttrs operator_attrs = operator_args.first; in TEST_F() local
143 std::string arg0_name = operator_attrs.at(0).first; in TEST_F()
144 ValuePtr arg0_value = operator_attrs.at(0).second; in TEST_F()
147 std::string arg1_name = operator_attrs.at(1).first; in TEST_F()
148 ValuePtr arg1_value = operator_attrs.at(1).second; in TEST_F()
178 OperatorAttrs operator_attrs = operator_args.first; in TEST_F() local
180 std::string arg0_name = operator_attrs.at(0).first; in TEST_F()
181 ValuePtr arg0_value = operator_attrs.at(0).second; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.cc300 OperatorAttrs operator_attrs; in CreateVirtualDivOp() local
301 operator_attrs.push_back(attr0); in CreateVirtualDivOp()
304 OperatorArgs operator_arg = std::make_pair(operator_attrs, operator_param); in CreateVirtualDivOp()
315 OperatorAttrs operator_attrs; in CreateReduceCommunicationOpArgs() local
316 operator_attrs.push_back(attr0); in CreateReduceCommunicationOpArgs()
317 operator_attrs.push_back(attr1); in CreateReduceCommunicationOpArgs()
320 return std::make_pair(operator_attrs, operator_param); in CreateReduceCommunicationOpArgs()
403 OperatorAttrs operator_attrs; in CreateAllGatherOp() local
404 operator_attrs.push_back(attr0); in CreateAllGatherOp()
407 OperatorArgs operator_arg = std::make_pair(operator_attrs, operator_param); in CreateAllGatherOp()
[all …]
Dgather_v2_info.cc207 OperatorAttrs operator_attrs; in CreateSubOp() local
216 OperatorArgs operator_args = std::make_pair(operator_attrs, operator_param); in CreateSubOp()