- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for source_tensor_ids (0.21 sec)
-
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) -
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)