Exemplo n.º 1
0
def _MaxPoolGradGradV2(op, grad):
    ksize = op.inputs[3]
    strides = op.inputs[4]
    return (array_ops.zeros_like(op.inputs[0]),
            array_ops.zeros_like(op.inputs[1]),
            gen_nn_ops.max_pool_grad_grad_v2(
                op.inputs[0],
                op.inputs[1],
                grad,
                ksize,
                strides,
                padding=op.get_attr("padding"),
                data_format=op.get_attr("data_format")), None, None)
Exemplo n.º 2
0
def _MaxPoolGradGradV2(op, grad):
  ksize = op.inputs[3]
  strides = op.inputs[4]
  return (array_ops.zeros(
      shape=array_ops.shape(op.inputs[0]), dtype=op.inputs[0].dtype),
          array_ops.zeros(
              shape=array_ops.shape(op.inputs[1]), dtype=op.inputs[1].dtype),
          gen_nn_ops.max_pool_grad_grad_v2(
              op.inputs[0],
              op.inputs[1],
              grad,
              ksize,
              strides,
              padding=op.get_attr("padding"),
              data_format=op.get_attr("data_format")), None, None)