import os
os.environ['TF ENABLE ONEDNN OPTS'] = '0'
import tensorflow as tf
print(tf. version )
with tf.device("CPU"):
  ksize = [1, 40, 128, 1]
  strides = [1, 128, 128, 30]
  padding = "SAME"
  data format = "NHWC"
  orig input shape = [11, 9, 78, 9]
  grad = tf.saturate cast(tf.random.uniform([16, 16, 16, 16], minval=-128, maxval=129, dtype=tf.int64), dtype=tf.float32)
  res = tf.raw ops.AvgPoolGrad(
    ksize=ksize,
    strides=strides,
    padding=padding,
    data format=data format,
    orig input shape=orig input shape,
    grad=grad,
  )