Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for gray (0.19 sec)

  1. tensorflow/c/c_api.h

    // Adds a copy of function `func` and optionally its gradient function `grad`
    // to `g`. Once `func`/`grad` is added to `g`, it can be called by creating
    // an operation using the function's name.
    // Any changes to `func`/`grad` (including deleting it) done after this method
    // returns, won't affect the copy of `func`/`grad` in `g`.
    // If `func` or `grad` are already in `g`, TF_GraphCopyFunction has no
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  2. tensorflow/c/eager/tape.h

        int64_t tensor_id = source_tensor_ids[i];
        auto grad_it = gradients.find(tensor_id);
        if (grad_it == gradients.end()) {
          result[i] = nullptr;
        } else {
          if (grad_it->second.size() > 1) {
            Gradient* grad = vspace.AggregateGradients(grad_it->second);
            grad_it->second.clear();
            grad_it->second.push_back(grad);
          }
          result[i] = grad_it->second[0];
          used_gradient_ids.insert(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