Searched refs:shape_element (Results 1 – 7 of 7) sorted by relevance
37 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape() local38 MS_EXCEPTION_IF_NULL(shape_element); in InferShape()39 return shape_element; in InferShape()
25 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape() local26 MS_EXCEPTION_IF_NULL(shape_element); in InferShape()27 return shape_element; in InferShape()
38 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape() local39 MS_EXCEPTION_IF_NULL(shape_element); in InferShape()40 return shape_element; in InferShape()
33 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape() local34 MS_EXCEPTION_IF_NULL(shape_element); in InferShape()35 return shape_element; in InferShape()
43 auto shape_element = x->cast<abstract::ShapePtr>(); in InferShape() local44 MS_EXCEPTION_IF_NULL(shape_element); in InferShape()45 return shape_element; in InferShape()
408 auto shape_element = shape->cast<abstract::ShapePtr>(); in ConvertShapePtrToShapeMap() local409 MS_EXCEPTION_IF_NULL(shape_element); in ConvertShapePtrToShapeMap()411 shape_map[kShape] = shape_element->shape(); in ConvertShapePtrToShapeMap()412 shape_map[kMinShape] = shape_element->min_shape(); in ConvertShapePtrToShapeMap()413 shape_map[kMaxShape] = shape_element->max_shape(); in ConvertShapePtrToShapeMap()