Searched refs:InferCallShape (Results 1 – 4 of 4) sorted by relevance
270 static StatusOr<Shape> InferCallShape(
1489 ShapeInference::InferCallShape({}, ShapeUtil::MakeProgramShape({}, f32_)); in TEST_F()1493 auto inferred_status1 = ShapeInference::InferCallShape( in TEST_F()1501 auto inferred_status_error0 = ShapeInference::InferCallShape( in TEST_F()1507 auto inferred_status_error1 = ShapeInference::InferCallShape( in TEST_F()1513 auto inferred_status_error2 = ShapeInference::InferCallShape( in TEST_F()
2833 /* static */ StatusOr<Shape> ShapeInference::InferCallShape( in InferCallShape() function in xla::ShapeInference
621 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferCallShape( in Call()