(op, grad, _)
return [op_module.ps_roi_align_grad(inputs_features, rois, grad, pooled_index, grid_dim_width, grid_dim_height, pool_method), None]
The gradients for `PsRoiAlign`.
The gradients for `PsRoiAlign`.
The gradients for PsRoiAlign .
def _ps_roi_align_grad(op, grad, _):
  '''The gradients for `PsRoiAlign`.
  '''
  inputs_features = op.inputs[0]
  rois = op.inputs[1]
  pooled_features_grad = op.outputs[0]
  pooled_index = op.outputs[1]
  grid_dim_width = op.get_attr('grid_dim_width')
  grid_dim_height = op.get_attr('grid_dim_height')

  #return [tf.ones_like(inputs_features), None]
  return [op_module.ps_roi_align_grad(inputs_features, rois, grad, pooled_index, grid_dim_width, grid_dim_height, pool_method), None]
def _ps_roi_align_grad ( op , grad , _ ) : inputs_features = op . inputs [ 0 ] rois = op . inputs [ 1 ] pooled_features_grad = op . outputs [ 0 ] pooled_index = op . outputs [ 1 ] grid_dim_width = op . get_attr ( 'grid_dim_width' ) grid_dim_height = op . get_attr ( 'grid_dim_height' ) #return [tf.ones_like(inputs_features), None] return [ op_module . ps_roi_align_grad ( inputs_features , rois , grad , pooled_index , grid_dim_width , grid_dim_height , pool_method ) , None ]