Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for NewMirrorPadGrad (0.18 sec)

  1. tensorflow/compiler/mlir/tfr/examples/pad/ops_defs.py

    
    @tf.RegisterGradient('NewMirrorPad')
    def _mirror_pad_grad(op, grad):
      mode = op.get_attr('mode')
      return [gen_array_ops.mirror_pad_grad(grad, op.inputs[1], mode=mode), None]
    
    
    @Composite(
        'NewMirrorPadGrad',
        inputs=['input_: T', 'paddings: Tpaddings'],
        attrs=['mode: {"REFLECT", "SYMMETRIC"}'],
        derived_attrs=['T: type', 'Tpaddings: {int32, int64} = DT_INT32'],
        outputs=['output: T'])
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 01 05:00:29 UTC 2021
    - 5.6K bytes
    - Viewed (0)
Back to top