HomeSort by relevance Sort by last modified time
    Searched refs:InfeedManager (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
infeed_manager.cc 27 InfeedManager::InfeedManager() : host_to_device_executor_(nullptr) {}
29 void InfeedManager::Reset() {
38 void InfeedManager::EnqueueBuffers(const std::vector<InfeedBuffer*>& buffers) {
53 InfeedBuffer* InfeedManager::BlockingDequeueBuffer() {
64 void InfeedManager::ReleaseBuffers(const std::vector<InfeedBuffer*>& buffers) {
77 se::Stream* InfeedManager::GetStream(se::StreamExecutor* executor) {
93 InfeedManager* GetOrCreateInfeedManager() {
94 static InfeedManager* manager = new InfeedManager;
    [all...]
infeed_manager.h 74 class InfeedManager {
76 InfeedManager();
87 // InfeedManager, either as a result of a call to Reset or because the
131 InfeedManager* GetOrCreateInfeedManager();
gpu_transfer_manager.cc 102 gpu::InfeedManager* infeed_manager = gpu::GetOrCreateInfeedManager();
135 gpu::InfeedManager* infeed_manager = gpu::GetOrCreateInfeedManager();
infeed_thunk.cc 40 InfeedManager* infeed_manager = GetOrCreateInfeedManager();

Completed in 856 milliseconds