Home
last modified time | relevance | path

Searched refs:InferWhileShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h226 static StatusOr<Shape> InferWhileShape(const ProgramShape& condition,
Dshape_inference_test.cc1945 ShapeInference::InferWhileShape(cond, body, result_shape); in TEST_F()
1959 ShapeInference::InferWhileShape(bad_shape_1, body, result_shape); in TEST_F()
1967 ShapeInference::InferWhileShape(cond, bad_shape_2, result_shape); in TEST_F()
1974 ShapeInference::InferWhileShape(bad_shape_3, body, result_shape); in TEST_F()
1981 ShapeInference::InferWhileShape(cond, bad_shape_4, result_shape); in TEST_F()
Dshape_inference.cc2746 /* static */ StatusOr<Shape> ShapeInference::InferWhileShape( in InferWhileShape() function in xla::ShapeInference
Dhlo_parser.cc1443 return ShapeInference::InferWhileShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2251 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferWhileShape( in While()