- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for SplitVGrad (0.15 sec)
-
tensorflow/cc/gradients/array_grad.cc
grad_outputs->push_back(NoGradient()); grad_outputs->push_back(Concat(scope, grad_inputs, op.input(0))); return scope.status(); } REGISTER_GRADIENT_OP("Split", SplitGrad); Status SplitVGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) { if (op.num_inputs() < 3) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
auto y = Split(scope_, split_dim, x, /* num_split */ 2); TensorShape y_shape = TensorShape({5, 1}); RunTest({x}, {x_shape}, y.output, {y_shape, y_shape}); } TEST_F(ArrayGradTest, SplitVGrad) { TensorShape x_shape({2, 6}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); auto y = SplitV(scope_, x, {1, 2, 3}, /*axis=*/1, /*num_split=*/3); RunTest({x}, {x_shape}, y.output,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0)