Searched refs:InferWhileShape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 226 static StatusOr<Shape> InferWhileShape(const ProgramShape& condition,
|
| D | shape_inference_test.cc | 1945 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()
|
| D | shape_inference.cc | 2746 /* static */ StatusOr<Shape> ShapeInference::InferWhileShape( in InferWhileShape() function in xla::ShapeInference
|
| D | hlo_parser.cc | 1443 return ShapeInference::InferWhileShape( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2251 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferWhileShape( in While()
|