Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for mirror_pad_grad (0.74 sec)

  1. 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)
  2. 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)
Back to top