Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for source_tensor_ids (0.21 sec)

  1. tensorflow/c/eager/tape.h

      }
      if (result.size() != source_tensor_ids.size()) {
        return errors::Internal("Expected result Span to be of size ",
                                source_tensor_ids.size(), " found ", result.size(),
                                " in call to Tape::ComputeGradient.");
      }
      std::unordered_set<int64_t> used_gradient_ids(source_tensor_ids.size());
      for (int i = 0; i < source_tensor_ids.size(); i++) {
    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)
  2. tensorflow/c/eager/gradients.cc

      TapeVSpace vspace(ctx);
      std::vector<int64_t> target_tensor_ids = MakeTensorIDList(targets);
      std::vector<int64_t> source_tensor_ids = MakeTensorIDList(sources);
      tensorflow::gtl::FlatSet<int64_t> sources_set(source_tensor_ids.begin(),
                                                    source_tensor_ids.end());
      std::unordered_map<int64_t, TapeTensor> sources_that_are_targets;
      for (int i = 0; i < target_tensor_ids.size(); ++i) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
Back to top