Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ReduceSum (0.08 sec)

  1. tensorflow/cc/gradients/array_grad.cc

      // all the gradients from the shape it fills.
      // We use ReduceSum to implement this, which needs an argument providing
      // the indices of all the dimensions of the incoming gradient.
      // grad(x) = reduce_sum(grad(y), [0..rank(grad(y))])
      auto all_dims = Range(scope, Const(scope, 0), Rank(scope, grad_inputs[0]),
                            Const(scope, 1));
      grad_outputs->push_back(ReduceSum(scope, grad_inputs[0], all_dims));
      return scope.status();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 31.7K bytes
    - Viewed (0)
  2. tensorflow/cc/gradients/linalg_grad.cc

          scope, Slice1dHelper(scope, x_shape, bx_start, bx_end),
          Slice1dHelper(scope, y_shape, by_start, by_end));
      grad_x = Reshape(
          scope, ReduceSum(scope, grad_x, Add(scope, bx_start, args.r0)), x_shape);
      grad_y = Reshape(
          scope, ReduceSum(scope, grad_y, Add(scope, by_start, args.r1)), y_shape);
      grad_outputs->push_back(grad_x);
      grad_outputs->push_back(grad_y);
      return scope.status();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 07 23:11:54 UTC 2022
    - 20.4K bytes
    - Viewed (0)
Back to top