HomeSort by relevance Sort by last modified time
    Searched refs:InfeedBuffer (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
infeed_manager.h 45 class InfeedBuffer {
47 InfeedBuffer() = default;
48 InfeedBuffer(se::StreamExecutor* executor, int64 length)
64 class InfeedManager : public XfeedQueue<ShapeTree<InfeedBuffer>> {
gpu_transfer_manager.h 48 // Initiates the infeed data transfers. InfeedBuffer->Done() must be
49 // called to clean up the memory allocated for InfeedBuffer.
50 StatusOr<InfeedBuffer> TransferBufferToInfeedInternal(
56 ShapeTree<InfeedBuffer> buffers);
gpu_transfer_manager.cc 57 ShapeTree<InfeedBuffer> buffer_tree(shape);
75 se::StreamExecutor* executor, ShapeTree<InfeedBuffer> buffers) {
96 StatusOr<InfeedBuffer> GpuTransferManager::TransferBufferToInfeedInternal(
112 InfeedBuffer buffer(executor, size);
infeed_thunk.cc 36 ShapeTree<InfeedBuffer> infeed_buffers =
69 InfeedBuffer* buffer =

Completed in 265 milliseconds