- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for RollGrad (0.32 sec)
-
tensorflow/cc/gradients/manip_grad.cc
#include "tensorflow/cc/framework/gradients.h" #include "tensorflow/cc/ops/manip_ops.h" #include "tensorflow/cc/ops/standard_ops.h" namespace tensorflow { namespace ops { namespace { Status RollGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) { auto shift = op.input(1); auto axis = op.input(2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 19 12:19:42 UTC 2020 - 1.4K bytes - Viewed (0) -
tensorflow/cc/gradients/manip_grad_test.cc
TF_ASSERT_OK((ComputeGradientError<float, float, float>( scope_, {x}, {x_shape}, {y}, {y_shape}, &max_error))); EXPECT_LT(max_error, 1e-4); } Scope scope_; }; TEST_F(ManipGradTest, RollGrad) { TensorShape shape({5, 4, 3}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape)); auto y = Roll(scope_, x, {2, 1}, {0, 1}); RunTest(x, shape, y, shape); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 19 12:19:42 UTC 2020 - 1.6K bytes - Viewed (0)