Searched refs:TuplePtr (Results 1 – 25 of 28) sorted by relevance
12
32 auto inputs_type = input_args[0]->BuildType()->cast<TuplePtr>()->elements(); in MergeInfer()
71 auto inputs_type = input_args[1]->BuildType()->cast<TuplePtr>()->elements(); in AssertInfer()
64 TuplePtr CTCLossV2InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &inpu… in CTCLossV2InferType()
63 auto x_types = input_args[0]->BuildType()->cast<TuplePtr>()->elements(); in PackInfer()
72 TuplePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
93 TuplePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
81 TuplePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
59 TuplePtr InferType(const PrimitivePtr &prim, const std::vector<AbstractBasePtr> &input_args) { in InferType()
59 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
60 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
61 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
62 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
63 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
72 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
70 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
68 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
73 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
129 using TuplePtr = std::shared_ptr<Tuple>; variable
55 auto tuple_type = cnode_type->cast<TuplePtr>(); in GenNewTensor()
207 auto tuple_output_type = primary_output_type->cast<mindspore::TuplePtr>(); in ExtractOutputTypeByNode()
396 auto type_tuple = type->cast<TuplePtr>(); in MakeAbstract()
65 TuplePtr tuple_type = dyn_cast<Tuple>(type); in SetTupleTypeProto()
147 TuplePtr tuple_type = dyn_cast<Tuple>(type); in SetNodeOutputType()