Searched refs:InferGetDimensionSizeShape (Results 1 – 6 of 6) sorted by relevance
327 static StatusOr<Shape> InferGetDimensionSizeShape(const Shape& shape,
162 ShapeInference::InferGetDimensionSizeShape( in ReplaceGetSize()
1163 ShapeInference::InferGetDimensionSizeShape( in HandleGetDimensionSize()
2303 /* static */ StatusOr<Shape> ShapeInference::InferGetDimensionSizeShape( in InferGetDimensionSizeShape() function in xla::ShapeInference
2571 return ShapeInference::InferGetDimensionSizeShape( in ParseInstructionRhs()
3330 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferGetDimensionSizeShape( in GetDimensionSize()