Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for mirror_pad_grad (0.22 sec)

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

        input_ = tf.raw_ops.Concat(
            concat_dim=i, values=[left_padding, input_, right_padding])
      return input_
    
    
    @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"}'],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 01 05:00:29 UTC 2021
    - 5.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/examples/pad/pad_ops_test.py

        kwargs_ = {
            'input_': input_,
            'paddings': paddings,
            'mode': mode,
        }
        # Make sure the composition python function is correct
        self._assertOpAndComposite([input_], tf.raw_ops.MirrorPadGrad,
                                   ops_defs._composite_mirror_pad_grad, kwargs_,
                                   kwargs)
        # Make sure the translation and decomposition is correct
        self._assertOpAndComposite([input_],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Sep 28 21:37:05 UTC 2021
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/cc/gradients/array_grad_test.cc

      TensorShape y_shape({2, 3});
      auto y = MirrorPadGrad(scope_, x, paddings, "REFLECT");
      RunTest(x, x_shape, y, y_shape);
    }
    
    TEST_F(ArrayGradTest, MirrorPadGradGrad_Symmetric) {
      TensorShape x_shape({4, 7});
      auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape));
      auto paddings = Const(scope_, {{1, 1}, {2, 2}});
      TensorShape y_shape({2, 3});
      auto y = MirrorPadGrad(scope_, x, paddings, "SYMMETRIC");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 19.3K bytes
    - Viewed (0)
  4. tensorflow/cc/gradients/array_grad.cc

    REGISTER_GRADIENT_OP("DepthToSpace", DepthToSpaceGrad);
    
    Status MirrorPadGrad(const Scope& scope, const Operation& op,
                         const std::vector<Output>& grad_inputs,
                         std::vector<Output>* grad_outputs) {
      string mode;
      TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "mode", &mode));
      grad_outputs->push_back(tensorflow::ops::internal::MirrorPadGrad(
          scope, grad_inputs[0], op.input(1), mode));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 31.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  6. RELEASE.md

    *   Fixes an overflow in `CONV_3D_TRANSPOSE` on TFLite ([CVE-2022-41894](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2022-41894))
    *   Fixes a heap OOB in `MirrorPadGrad` ([CVE-2022-41895](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2022-41895))
    *   Fixes a crash in `Mfcc` ([CVE-2022-41896](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2022-41896))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 730.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
      TF_DerivedOperandTypeAttr Tpaddings = TF_DerivedOperandTypeAttr<1>;
    }
    
    def TF_MirrorPadGradOp : TF_Op<"MirrorPadGrad", [Pure, TF_OperandHasRank<1, 2>]> {
      let summary = [{
    Gradient op for `MirrorPad` op. This op folds a mirror-padded tensor.
      }];
    
      let description = [{
    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