Searched refs:InferSelectAndScatterShape (Results 1 – 5 of 5) sorted by relevance
164 static StatusOr<Shape> InferSelectAndScatterShape(
328 auto inferred_status_ok = ShapeInference::InferSelectAndScatterShape( in TEST_F()338 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()349 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()360 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()371 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()382 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()
639 ShapeInference::InferSelectAndScatterShape( in HandleSelectAndScatter()
2124 /* static */ StatusOr<Shape> ShapeInference::InferSelectAndScatterShape( in InferSelectAndScatterShape() function in xla::ShapeInference
2281 ShapeInference::InferSelectAndScatterShape( in SelectAndScatterWithGeneralPadding()