Searched refs:InferWhileShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 207 static StatusOr<Shape> InferWhileShape(const ProgramShape& condition,
|
| D | shape_inference_test.cc | 1310 ShapeInference::InferWhileShape(cond, body, result_shape); in TEST_F()
1324 ShapeInference::InferWhileShape(bad_shape_1, body, result_shape); in TEST_F()
1332 ShapeInference::InferWhileShape(cond, bad_shape_2, result_shape); in TEST_F()
1339 ShapeInference::InferWhileShape(bad_shape_3, body, result_shape); in TEST_F()
1346 ShapeInference::InferWhileShape(cond, bad_shape_4, result_shape); in TEST_F()
|
| D | shape_inference.cc | 2519 /* static */ StatusOr<Shape> ShapeInference::InferWhileShape( in InferWhileShape() function in xla::ShapeInference
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1829 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferWhileShape( in While()
|