- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for mirror_pad_grad (0.74 sec)
-
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) -
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)