Searched refs:InferConditionalShape (Results 1 – 4 of 4) sorted by relevance
1541 auto inferred_status0 = ShapeInference::InferConditionalShape( in TEST_F()1549 auto inferred_status1 = ShapeInference::InferConditionalShape( in TEST_F()1558 auto inferred_status2 = ShapeInference::InferConditionalShape( in TEST_F()1566 auto inferred_status_error0 = ShapeInference::InferConditionalShape( in TEST_F()1575 auto inferred_status_error1 = ShapeInference::InferConditionalShape( in TEST_F()1584 auto inferred_status_error2 = ShapeInference::InferConditionalShape( in TEST_F()1594 auto inferred_status_error3 = ShapeInference::InferConditionalShape( in TEST_F()1603 auto inferred_status_error4 = ShapeInference::InferConditionalShape( in TEST_F()1613 auto inferred_status_error5 = ShapeInference::InferConditionalShape( in TEST_F()1626 auto inferred_status0 = ShapeInference::InferConditionalShape( in TEST_F()[all …]
212 static StatusOr<Shape> InferConditionalShape(
2555 /* static */ StatusOr<Shape> ShapeInference::InferConditionalShape( in InferConditionalShape() function in xla::ShapeInference
1920 ShapeInference::InferConditionalShape( in Conditional()