Searched refs:InferSelectAndScatterShape (Results 1 – 7 of 7) sorted by relevance
214 static StatusOr<Shape> InferSelectAndScatterShape(
342 auto inferred_status_ok = ShapeInference::InferSelectAndScatterShape( in TEST_F()352 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()363 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()374 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()385 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()396 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()
1283 ShapeInference::InferSelectAndScatterShape( in HandleSelectAndScatter()
2351 /* static */ StatusOr<Shape> ShapeInference::InferSelectAndScatterShape( in InferSelectAndScatterShape() function in xla::ShapeInference
2113 ShapeInference::InferSelectAndScatterShape( in Propagate()
2253 return ShapeInference::InferSelectAndScatterShape( in CreateInstruction()
3321 ShapeInference::InferSelectAndScatterShape( in SelectAndScatterInternal()