Searched refs:EnqueueBuffersAtomically (Results 1 – 4 of 4) sorted by relevance
90 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F()91 xfeed->infeed()->EnqueueBuffersAtomically({b}); in TEST_F()102 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F()104 xfeed->infeed()->EnqueueBuffersAtomically({b}); in TEST_F()125 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F()134 xfeed->outfeed()->EnqueueBuffersAtomically({b}); in TEST_F()142 xfeed->outfeed()->EnqueueBuffersAtomically({b}); in TEST_F()150 xfeed->outfeed()->EnqueueBuffersAtomically({b}); in TEST_F()
133 xfeed_manager->infeed()->EnqueueBuffersAtomically(buffers); in TransferLiteralToInfeed()147 xfeed_manager->infeed()->EnqueueBuffersAtomically({buffer}); in TransferBufferToInfeed()272 xfeed_manager->outfeed()->EnqueueBuffersAtomically(buffer_pointers); in TransferBuffersFromOutfeedInternal()
67 void EnqueueBuffersAtomically(absl::Span<XfeedBuffer* const> buffers);
39 void XfeedQueueManager::EnqueueBuffersAtomically( in EnqueueBuffersAtomically() function in xla::cpu::runtime::XfeedQueueManager