Searched refs:InfeedBuffer (Results 1 – 4 of 4) sorted by relevance
45 class InfeedBuffer {47 InfeedBuffer() = default;48 InfeedBuffer(se::StreamExecutor* executor, int64 length) in InfeedBuffer() function64 class InfeedManager : public XfeedQueue<ShapeTree<InfeedBuffer>> {
57 ShapeTree<InfeedBuffer> buffer_tree(shape); in TransferLiteralToInfeed()75 se::StreamExecutor* executor, ShapeTree<InfeedBuffer> buffers) { in EnqueueBuffersToInfeed()96 StatusOr<InfeedBuffer> GpuTransferManager::TransferBufferToInfeedInternal( in TransferBufferToInfeedInternal()112 InfeedBuffer buffer(executor, size); in TransferBufferToInfeedInternal()
50 StatusOr<InfeedBuffer> TransferBufferToInfeedInternal(56 ShapeTree<InfeedBuffer> buffers);
36 ShapeTree<InfeedBuffer> infeed_buffers = in ExecuteOnStream()69 InfeedBuffer* buffer = in ExecuteOnStream()