Searched refs:InferReduceWindowShape (Results 1 – 12 of 12) sorted by relevance
197 static StatusOr<Shape> InferReduceWindowShape(200 static StatusOr<Shape> InferReduceWindowShape(const Shape& operand_shape,203 static StatusOr<Shape> InferReduceWindowShape(208 static StatusOr<Shape> InferReduceWindowShape(
94 ShapeInference::InferReduceWindowShape( in HandleReduce()
2293 /* static */ StatusOr<Shape> ShapeInference::InferReduceWindowShape( in InferReduceWindowShape() function in xla::ShapeInference2299 return InferReduceWindowShape(operand_shape, init_value_shape, window); in InferReduceWindowShape()2302 /* static */ StatusOr<Shape> ShapeInference::InferReduceWindowShape( in InferReduceWindowShape() function in xla::ShapeInference2332 InferReduceWindowShape(*operands[i], *init_values[i], window)); in InferReduceWindowShape()2343 /* static */ StatusOr<Shape> ShapeInference::InferReduceWindowShape( in InferReduceWindowShape() function in xla::ShapeInference
333 auto inferred_status = ShapeInference::InferReduceWindowShape( in TEST_F()1153 auto inferred_status = ShapeInference::InferReduceWindowShape( in TEST_F()1215 auto inferred_status = ShapeInference::InferReduceWindowShape( in TEST_F()
2942 ShapeInference::InferReduceWindowShape( in TEST_P()2993 ShapeInference::InferReduceWindowShape( in TEST_P()
1270 reduce_window, ShapeInference::InferReduceWindowShape( in HandleReduceWindow()
1764 ShapeInference::InferReduceWindowShape(
2259 ShapeInference::InferReduceWindowShape( in Propagate()
1967 return ShapeInference::InferReduceWindowShape( in CreateInstruction()
5759 ShapeInference::InferReduceWindowShape( in TEST_P()
2728 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferReduceWindowShape( in ReduceWindowWithGeneralPadding()2773 ShapeInference::InferReduceWindowShape( in ReduceWindowInternal()
3925 ShapeInference::InferReduceWindowShape( in HandleReduceWindow()