예제 #1
0
def _AvgPool3DGrad(op, grad):
    return gen_nn_ops.avg_pool3d_grad(array_ops.shape(op.inputs[0]),
                                      grad,
                                      ksize=op.get_attr("ksize"),
                                      strides=op.get_attr("strides"),
                                      padding=op.get_attr("padding"),
                                      data_format=op.get_attr("data_format"))
def _AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding):
    del outputs  # Unused by average-pooling gradients.
    return gen_nn_ops.avg_pool3d_grad(inputs.get_shape().as_list(),
                                      output_gradients,
                                      ksize=ksize,
                                      strides=strides,
                                      padding=padding)
예제 #3
0
def _AvgPool3DGrad(op, grad):
  return gen_nn_ops.avg_pool3d_grad(
      array_ops.shape(op.inputs[0]),
      grad,
      ksize=op.get_attr("ksize"),
      strides=op.get_attr("strides"),
      padding=op.get_attr("padding"),
      data_format=op.get_attr("data_format").decode())
예제 #4
0
def _AvgPoolGrad(inputs, outputs, output_gradients, ksize, strides, padding):
  del outputs  # Unused by average-pooling gradients.
  return gen_nn_ops.avg_pool3d_grad(
      inputs.get_shape().as_list(),
      output_gradients,
      ksize=ksize,
      strides=strides,
      padding=padding)