Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for UnsortedSegmentSum (0.12 sec)

  1. tensorflow/cc/gradients/math_grad.cc

                                  std::vector<Output>* grad_outputs) {
      if (op.num_inputs() != 3) {
        return errors::InvalidArgument("UnsortedSegmentSum requires 3 arguments");
      }
    
      if (grad_inputs.size() != 1) {
        return errors::InvalidArgument(
            "UnsortedSegmentSum grad requires 1 grad input");
      }
    
      auto zero_clipped_indices = GetZeroClippedIndices(scope, op.input(1));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top