Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for IDs (0.33 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top