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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
xfeed_manager.h 64 // called when the buffer will no longer be accessed by the XfeedManager,
107 class XfeedManager {
109 XfeedManager() = default;
cpu_runtime.cc 33 XfeedManager* GetXfeedManager(int device_ordinal) {
34 static auto* managers = new absl::flat_hash_map<int, XfeedManager*>();
40 it = managers->emplace(device_ordinal, new XfeedManager()).first;
118 xla::cpu::runtime::XfeedManager* xfeed =
142 xla::cpu::runtime::XfeedManager* xfeed =
161 xla::cpu::runtime::XfeedManager* xfeed =
185 xla::cpu::runtime::XfeedManager* xfeed =
cpu_runtime.h 75 XfeedManager* GetXfeedManager(int device_ordinal);
xfeed_manager_test.cc 88 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0);
100 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0);
111 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0);
133 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0);
xfeed_manager.cc 25 void XfeedManager::Reset() {
cpu_transfer_manager.cc 131 cpu::runtime::XfeedManager* xfeed_manager =
145 cpu::runtime::XfeedManager* xfeed_manager =
270 cpu::runtime::XfeedManager* xfeed_manager =

Completed in 241 milliseconds