Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for grad_index (0.41 sec)

  1. tensorflow/c/eager/tape.h

      targets.reserve(grad.size());
      used_in_grads.reserve(grad.size());
      std::unordered_map<int64_t, TapeTensor> sources_that_are_targets;
      for (int grad_index = 0, end = grad.size(); grad_index < end; ++grad_index) {
        Gradient* grad_tensor = grad[grad_index];
        if (grad_tensor != nullptr) {
          int64_t tensor_id = vspace_.TensorId(grad_tensor);
          targets.push_back(tensor_id);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
Back to top