Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for pack (0.12 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_testlib.h

                                            TF_Status* status);
    
    // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle.
    template <std::size_t num_replicas>
    void ExtractPerDeviceValues(
        TFE_Context* context, TFE_TensorHandle* input,
        std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status);
    
    // Helper to pack `num_replicas` TFE_TensorHandles into one parallel handle.
    template <std::size_t num_replicas>
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 09 01:12:35 GMT 2021
    - 6.9K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/filesystem/plugins/gcs/expiring_lru_cache.h

          lru_list_.erase(insert.first->second.lru_iterator);
          insert.first->second = entry;
        } else if (max_entries_ > 0 && cache_.size() > max_entries_) {
          cache_.erase(lru_list_.back());
          lru_list_.pop_back();
        }
      }
    
      bool DeleteLocked(const std::string& key) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_) {
        auto it = cache_.find(key);
        if (it == cache_.end()) {
          return false;
        }
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 09 19:31:22 GMT 2020
    - 6.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device.h

    // parallel device creates a parallel tensor `x` with `a` on the first of
    // `underlying_devices` and `b` on the second. Running `a_unpacked, b_unpacked =
    // TPUReplicatedOutput(input=x, num_replicas=2)` un-packs the parallel tensor
    // into its components.
    //
    // The filled `device` struct and the allocated `device_info` struct may be
    // passed to TFE_RegisterCustomDevice. The `device_name` arguments must match.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jun 04 21:49:16 GMT 2020
    - 2.9K bytes
    - Viewed (0)
  4. tensorflow/c/env.h

    // (for debugging/performance-analysis) by thread_name.
    //
    // The given param (which may be null) is passed to work_func when the thread
    // starts. In this way, data may be passed from the thread back to the caller.
    //
    // Caller takes ownership of the result and must call TF_JoinThread on it
    // eventually.
    TF_CAPI_EXPORT extern TF_Thread* TF_StartThread(const TF_ThreadOptions* options,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sat Jan 09 02:53:27 GMT 2021
    - 9.6K bytes
    - Viewed (0)
  5. tensorflow/c/eager/abstract_operation.h

      std::vector<const char*> raw_strs;
      std::vector<size_t> lengths;
      raw_strs.reserve(values.size());
      lengths.reserve(values.size());
      for (const auto& s : values) {
        raw_strs.emplace_back(s.data());
        lengths.emplace_back(s.size());
      }
      return SetAttrStringList(attr_name,
                               reinterpret_cast<const void**>(raw_strs.data()),
                               lengths.data(), values.size());
    }
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Jul 14 16:20:41 GMT 2021
    - 6.8K bytes
    - Viewed (0)
Back to top