import tensorflow as tf

@tf.function
def test():
 y=tf.raw ops.AvgPoolGrad(
  orig input shape=[1,1,1,1],
  grad=[[[[1.0],[1.0],[1.0]]],[[[2.0],[2.0],[2.0]]],[[[3.0],[3.0],[3.0]]]],
  ksize=[1,1,1,1],
  strides=[1,1,1,0],
  padding='VALID',
  data format='NCHW')
 return y

test()

int64 t GetOutputSize(const int64 t input, const int64 t filter,
           const int64 t stride, const Padding& padding) {
 // Logic for calculating output shape is from GetWindowedOutputSizeVerbose() 
 // function in third party/tensorflow/core/framework/common shape fns.cc.
 if (padding == Padding::VALID) {
  return (input - filter + stride) / stride;
 } else { // SAME.
  return (input + stride - 1) / stride;
 }
}