- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for IDs (0.33 sec)
-
tensorflow/c/eager/gradients.cc
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) { int64_t target_id = target_tensor_ids[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) -
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/parallel_device/parallel_device_lib.cc
std::unique_ptr<ParallelTensor> ParallelDevice::DeviceIDs( TFE_Context* context, TF_Status* status) const { std::vector<int32_t> ids; ids.reserve(num_underlying_devices()); for (int i = 0; i < num_underlying_devices(); ++i) { ids.push_back(i); } return ScalarsFromSequence<int32_t>(ids, context, status); } absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> ParallelDevice::Execute(TFE_Context* context,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1)