Home
last modified time | relevance | path

Searched refs:operator_args (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dmatmul_info_test.cc353 OperatorArgs operator_args = forward_op.at(0).second; in TEST_F() local
355 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
356 ValuePtr arg0_value = operator_args.first.at(0).second; in TEST_F()
358 std::string arg1_name = operator_args.first.at(1).first; in TEST_F()
359 ValuePtr arg1_value = operator_args.first.at(1).second; in TEST_F()
389 OperatorArgs operator_args = virtual_div_op.at(0).second; in TEST_F() local
391 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
392 ValuePtr arg0_value = operator_args.first.at(0).second; in TEST_F()
409 OperatorArgs operator_args = mirror_op.at(0).second; in TEST_F() local
411 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
[all …]
Dprelu_test.cc130 OperatorArgs operator_args = mirror_op.at(0).second; in TEST_F() local
131 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
132 ValuePtr arg0_value = operator_args.first.at(0).second; in TEST_F()
235 OperatorArgs operator_args = mirror_op.at(0).second; in TEST_F() local
236 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
237 ValuePtr arg0_value = operator_args.first.at(0).second; in TEST_F()
Dreduce_method_test.cc140 OperatorArgs operator_args = forward_op.at(0).second; in TEST_F() local
141 OperatorAttrs operator_attrs = operator_args.first; in TEST_F()
177 OperatorArgs operator_args = mirror_op.at(0).second; in TEST_F() local
178 OperatorAttrs operator_attrs = operator_args.first; in TEST_F()
Dactivation_info_test.cc139 OperatorArgs operator_args = mirror_op.at(0).second; in TEST_F() local
141 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
142 ValuePtr arg0_value = operator_args.first.at(0).second; in TEST_F()
Dl2_normalize_info_test.cc182 OperatorArgs operator_args = mirror_op.at(0).second; in TEST_F() local
183 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
184 ValuePtr arg0_value = operator_args.first.at(0).second; in TEST_F()
Dsoftmax_entropy_loss_info_test.cc165 OperatorArgs operator_args = virtual_div_op.at(0).second; in TEST_F() local
167 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
168 ValuePtr arg0_value = operator_args.first.at(0).second; in TEST_F()
Dtensor_add_info_test.cc234 OperatorArgs operator_args = mirror_op.at(0).second; in TEST_F() local
236 std::string arg0_name = operator_args.first.at(0).first; in TEST_F()
237 ValuePtr arg0_value = operator_args.first.at(0).second; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dgather_v2_info.cc216 OperatorArgs operator_args = std::make_pair(operator_attrs, operator_param); in CreateSubOp() local
217 Operator op = std::make_pair(operator_name, operator_args); in CreateSubOp()
Doperator_info.cc514 OperatorArgs operator_args = std::make_pair(operator_attrs, operator_param); in CreateMirrorOps() local
516 Operator op = std::make_pair(operator_name, operator_args); in CreateMirrorOps()