Searched refs:InferConditionalShape (Results 1 – 5 of 5) sorted by relevance
2190 auto inferred_status0 = ShapeInference::InferConditionalShape( in TEST_F()2198 auto inferred_status1 = ShapeInference::InferConditionalShape( in TEST_F()2207 auto inferred_status2 = ShapeInference::InferConditionalShape( in TEST_F()2215 auto inferred_status_error0 = ShapeInference::InferConditionalShape( in TEST_F()2224 auto inferred_status_error1 = ShapeInference::InferConditionalShape( in TEST_F()2233 auto inferred_status_error2 = ShapeInference::InferConditionalShape( in TEST_F()2243 auto inferred_status_error3 = ShapeInference::InferConditionalShape( in TEST_F()2252 auto inferred_status_error4 = ShapeInference::InferConditionalShape( in TEST_F()2262 auto inferred_status_error5 = ShapeInference::InferConditionalShape( in TEST_F()2275 auto inferred_status0 = ShapeInference::InferConditionalShape( in TEST_F()[all …]
231 static StatusOr<Shape> InferConditionalShape(
2783 /* static */ StatusOr<Shape> ShapeInference::InferConditionalShape( in InferConditionalShape() function in xla::ShapeInference
2223 return ShapeInference::InferConditionalShape( in ParseInstructionRhs()
2396 ShapeInference::InferConditionalShape( in ConditionalImpl()