Searched refs:InferSelectAndScatterShape (Results 1 – 7 of 7) sorted by relevance
183 static StatusOr<Shape> InferSelectAndScatterShape(
340 auto inferred_status_ok = ShapeInference::InferSelectAndScatterShape( in TEST_F()350 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()361 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()372 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()383 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()394 auto inferred_status_fail = ShapeInference::InferSelectAndScatterShape( in TEST_F()
913 ShapeInference::InferSelectAndScatterShape( in HandleSelectAndScatter()
1476 ShapeInference::InferSelectAndScatterShape( in Propagate()
2243 /* static */ StatusOr<Shape> ShapeInference::InferSelectAndScatterShape( in InferSelectAndScatterShape() function in xla::ShapeInference
1889 return ShapeInference::InferSelectAndScatterShape( in ParseInstructionRhs()
3091 ShapeInference::InferSelectAndScatterShape( in SelectAndScatterInternal()