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([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 …]
118 xla::cpu::runtime::XfeedManager* xfeed = in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() local122 xfeed->infeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()142 xla::cpu::runtime::XfeedManager* xfeed = in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() local146 xfeed->infeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()161 xla::cpu::runtime::XfeedManager* xfeed = in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() local165 xfeed->outfeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()185 xla::cpu::runtime::XfeedManager* xfeed = in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() local189 xfeed->outfeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()