Searched refs:xfeed (Results 1 – 2 of 2) sorted by relevance
88 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F() local90 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F()91 xfeed->infeed()->EnqueueBuffersAtomically({b}); in TEST_F()100 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F() local102 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F()104 xfeed->infeed()->EnqueueBuffersAtomically({b}); in TEST_F()111 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F() local115 pool.Schedule([length, &xfeed]() { in TEST_F()125 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F()133 cpu::runtime::XfeedManager* xfeed = cpu::runtime::GetXfeedManager(0); in TEST_F() local[all …]
242 xla::cpu::runtime::XfeedManager* xfeed = in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() local246 xfeed->infeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()266 xla::cpu::runtime::XfeedManager* xfeed = in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() local270 xfeed->infeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()285 xla::cpu::runtime::XfeedManager* xfeed = in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() local289 xfeed->outfeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()309 xla::cpu::runtime::XfeedManager* xfeed = in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() local313 xfeed->outfeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()