Searched refs:shape_tuple (Results 1 – 5 of 5) sorted by relevance
71 std::vector<abstract::BaseShapePtr> shape_tuple; in InferShape() local76 shape_tuple.push_back(out_shape); in InferShape()78 return std::make_shared<abstract::TupleShape>(shape_tuple); in InferShape()
127 std::shared_ptr<TupleShape> shape_tuple = dyn_cast<TupleShape>(abstract_tup->BuildShape()); in TEST_F() local128 ASSERT_TRUE(shape_tuple); in TEST_F()129 const std::vector<BaseShapePtr> &ptr_vec = shape_tuple->shape(); in TEST_F()
395 auto shape_tuple = base_shape->cast<TupleShapePtr>(); in MakeAbstract() local398 for (size_t it = 0; it < shape_tuple->size(); ++it) { in MakeAbstract()399 auto tensor_it = MakeAbstract((*shape_tuple)[it], (*type_tuple)[it]); in MakeAbstract()
466 auto shape_tuple = shape_obj.cast<py::tuple>(); in MakePyInferRes2Abstract() local469 for (size_t it = 0; it < shape_tuple.size(); ++it) { in MakePyInferRes2Abstract()470 auto tensor_it = MakePyInferRes2Abstract(shape_tuple[it], typeid_tuple[it]); in MakePyInferRes2Abstract()
309 py::tuple shape_tuple(len); in AbstractTupleToPython() local322 shape_tuple[i] = out[ATTR_SHAPE]; in AbstractTupleToPython()341 dic[ATTR_SHAPE] = shape_tuple; in AbstractTupleToPython()