Method: TensorStream::MathGradients._Conv2DGrad

Defined in:

lib/tensor_stream/math_gradients.rb

._Conv2DGrad(op, grad) ⇒ Object

# File 'lib/tensor_stream/math_gradients.rb', line 284

def self._Conv2DGrad(op, grad)
  # dilations = op.get_attr("dilations")
  strides = op.options[:strides]
  padding = op.options[:padding]
  use_cudnn_on_gpu = op.options[:use_cudnn_on_gpu]
  data_format = op.options[:data_format]

  shape_0, shape_1 = ts.shape_n([op.inputs[0], op.inputs[1]])
  [
    _op(:conv2d_backprop_input,
      shape_0,
      op.inputs[1],
      grad,
      strides: strides,
        padding: padding,
        use_cudnn_on_gpu: use_cudnn_on_gpu,
        data_format: data_format),
    _op(:conv2d_backprop_filter,
      op.inputs[0],
      shape_1,
      grad,
      strides: strides,
      padding: padding,
      use_cudnn_on_gpu: use_cudnn_on_gpu,
      data_format: data_format),
  ]
end