Searched refs:InferReduceWindowShape (Results 1 – 12 of 12) sorted by relevance
166 static StatusOr<Shape> InferReduceWindowShape(169 static StatusOr<Shape> InferReduceWindowShape(const Shape& operand_shape,172 static StatusOr<Shape> InferReduceWindowShape(177 static StatusOr<Shape> InferReduceWindowShape(
80 ShapeInference::InferReduceWindowShape( in HandleReduce()
2187 /* static */ StatusOr<Shape> ShapeInference::InferReduceWindowShape( in InferReduceWindowShape() function in xla::ShapeInference2193 return InferReduceWindowShape(operand_shape, init_value_shape, window); in InferReduceWindowShape()2196 /* static */ StatusOr<Shape> ShapeInference::InferReduceWindowShape( in InferReduceWindowShape() function in xla::ShapeInference2223 InferReduceWindowShape(*operands[i], *init_values[i], window)); in InferReduceWindowShape()2234 /* static */ StatusOr<Shape> ShapeInference::InferReduceWindowShape( in InferReduceWindowShape() function in xla::ShapeInference
331 auto inferred_status = ShapeInference::InferReduceWindowShape( in TEST_F()1118 auto inferred_status = ShapeInference::InferReduceWindowShape( in TEST_F()1180 auto inferred_status = ShapeInference::InferReduceWindowShape( in TEST_F()
900 reduce_window, ShapeInference::InferReduceWindowShape( in HandleReduceWindow()
2904 ShapeInference::InferReduceWindowShape( in TEST_P()2955 ShapeInference::InferReduceWindowShape( in TEST_P()
1622 ShapeInference::InferReduceWindowShape( in Propagate()
1951 ShapeInference::InferReduceWindowShape(
1589 return ShapeInference::InferReduceWindowShape( in ParseInstructionRhs()
5119 ShapeInference::InferReduceWindowShape( in TEST_P()
2574 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferReduceWindowShape( in ReduceWindowWithGeneralPadding()2619 ShapeInference::InferReduceWindowShape( in ReduceWindowInternal()
3122 ShapeInference::InferReduceWindowShape( in HandleReduceWindow()