Searched refs:InfeedBuffer (Results 1 – 4 of 4) sorted by relevance
46 class InfeedBuffer {48 InfeedBuffer() = default;49 InfeedBuffer(se::StreamExecutor* executor, int64 length) in InfeedBuffer() function65 class InfeedManager : public XfeedQueue<ShapeTree<InfeedBuffer>> {
57 ShapeTree<InfeedBuffer> buffer_tree(literal_shape); in TransferLiteralToInfeed()70 se::StreamExecutor* executor, ShapeTree<InfeedBuffer> buffers) { in EnqueueBuffersToInfeed()91 StatusOr<InfeedBuffer> GpuTransferManager::TransferBufferToInfeedInternal( in TransferBufferToInfeedInternal()108 InfeedBuffer buffer(executor, size); in TransferBufferToInfeedInternal()
49 StatusOr<InfeedBuffer> TransferBufferToInfeedInternal(55 ShapeTree<InfeedBuffer> buffers);
41 ShapeTree<InfeedBuffer> source_buffers = in ExecuteOnStream()48 InfeedBuffer& buffer = source.second; in ExecuteOnStream()