Home
last modified time | relevance | path

Searched refs:GetXfeedManager (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc33 XfeedManager* GetXfeedManager(int device_ordinal) { in GetXfeedManager() function
119 xla::cpu::runtime::GetXfeedManager(device_ordinal); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
143 xla::cpu::runtime::GetXfeedManager(device_ordinal); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
162 xla::cpu::runtime::GetXfeedManager(device_ordinal); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
186 xla::cpu::runtime::GetXfeedManager(device_ordinal); in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
Dxfeed_manager_test.cc88 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F()
100 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F()
111 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F()
133 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F()
Dcpu_transfer_manager.cc132 cpu::runtime::GetXfeedManager(executor->device_ordinal()); in TransferLiteralToInfeed()
146 cpu::runtime::GetXfeedManager(executor->device_ordinal()); in TransferBufferToInfeed()
271 cpu::runtime::GetXfeedManager(executor->device_ordinal()); in TransferBuffersFromOutfeedInternal()
Dcpu_runtime.h75 XfeedManager* GetXfeedManager(int device_ordinal);