Searched defs:element_shapes (Results 1 – 2 of 2) sorted by relevance
37 …eVector CheckAndGetInferredShape(const PrimitivePtr &primitive, const ShapeArray &element_shapes) { in CheckAndGetInferredShape()87 ShapeArray element_shapes; in InferShape() local97 ShapeArray element_shapes; in InferShape() local
2268 const auto &element_shapes = std::vector<abstract::BaseShapePtr>(values.size(), single_shape); in SequenceToTensor() local2281 const auto &element_shapes = std::vector<abstract::BaseShapePtr>(values.size(), element_shape); in SequenceToTensor() local