Searched refs:InferCallShape (Results 1 – 5 of 5) sorted by relevance
299 static StatusOr<Shape> InferCallShape(
2138 ShapeInference::InferCallShape({}, ShapeUtil::MakeProgramShape({}, f32_)); in TEST_F()2142 auto inferred_status1 = ShapeInference::InferCallShape( in TEST_F()2150 auto inferred_status_error0 = ShapeInference::InferCallShape( in TEST_F()2156 auto inferred_status_error1 = ShapeInference::InferCallShape( in TEST_F()2162 auto inferred_status_error2 = ShapeInference::InferCallShape( in TEST_F()
3178 /* static */ StatusOr<Shape> ShapeInference::InferCallShape( in InferCallShape() function in xla::ShapeInference
1560 return ShapeInference::InferCallShape( in ParseInstructionRhs()
833 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferCallShape( in Call()