Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for RegisterGradient (0.14 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    such that dx = g(dy). In Python,
    
    ```python
    with tf.get_default_graph().gradient_override_map(
        {'IdentityN': 'OverrideGradientWithG'}):
      y, _ = identity_n([f(x), x])
    
    @tf.RegisterGradient('OverrideGradientWithG')
    def ApplyG(op, dy, _):
      return [None, g(dy)]  # Do not backprop to f(x).
    ```
      }];
    
      let arguments = (ins
        Variadic<TF_Tensor>:$input
      );
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top