Searched refs:GetXfeedManager (Results 1 – 4 of 4) sorted by relevance
88 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()141 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F()149 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F()
132 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()
52 XfeedManager* GetXfeedManager(int device_ordinal) { in GetXfeedManager() function243 xla::cpu::runtime::GetXfeedManager(device_ordinal); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()267 xla::cpu::runtime::GetXfeedManager(device_ordinal); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()286 xla::cpu::runtime::GetXfeedManager(device_ordinal); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()310 xla::cpu::runtime::GetXfeedManager(device_ordinal); in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
89 XfeedManager* GetXfeedManager(int device_ordinal);