Searched refs:each_shape (Results 1 – 2 of 2) sorted by relevance
140 auto each_shape = dyn_cast<abstract::Shape>(shape); in GetNodeShape() local141 MS_EXCEPTION_IF_NULL(each_shape); in GetNodeShape()142 shapes.push_back(each_shape->shape()); in GetNodeShape()
155 std::vector<int64_t> each_shape = GetValue<std::vector<int64_t>>(each_shape_value); in InferShape() local156 BaseShapePtr base_shape = std::make_shared<abstract::Shape>(each_shape); in InferShape()