Home
last modified time | relevance | path

Searched refs:input_tuple (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_addn.cc40 auto input_tuple = std::make_shared<ValueTuple>(std::vector<ValuePtr>{tensor_x1, tensor_x2}); in TEST_F() local
41 auto abstract = addn->Infer({input_tuple->ToAbstract()}); in TEST_F()
73 …auto input_tuple = std::make_shared<ValueTuple>(std::vector<ValuePtr>{tensor_x1, tensor_x2, tensor… in TEST_F() local
74 auto abstract = addn->Infer({input_tuple->ToAbstract()}); in TEST_F()
Dtest_ops_concat.cc43 …auto input_tuple = std::make_shared<ValueTuple>(std::vector<ValuePtr>{tensor_x1, tensor_x2, tensor… in TEST_F() local
44 auto abstract = concat->Infer({input_tuple->ToAbstract()}); in TEST_F()
77 …auto input_tuple = std::make_shared<ValueTuple>(std::vector<ValuePtr>{tensor_x1, tensor_x2, tensor… in TEST_F() local
78 auto abstract = concat->Infer({input_tuple->ToAbstract()}); in TEST_F()
/third_party/mindspore/mindspore/core/ops/
Dconcat.cc40 auto input_tuple = input_args[0]->cast<abstract::AbstractTuplePtr>(); in ConcatInfer() local
41 MS_EXCEPTION_IF_NULL(input_tuple); in ConcatInfer()
42 auto elements = input_tuple->elements(); in ConcatInfer()
/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc1167 auto input_tuple = args_spec_list[0]->cast<abstract::AbstractSequeuePtr>(); in InferImplDynamicStitch() local
1168 MS_EXCEPTION_IF_NULL(input_tuple); in InferImplDynamicStitch()
1169 auto indices = input_tuple->elements(); in InferImplDynamicStitch()