Lines Matching refs:executor
49 Status GpuTransferManager::TransferLiteralToInfeed(se::StreamExecutor* executor,
57 return TransferBufferToInfeed(executor, size, literal.untyped_data());
83 TransferBufferToInfeedInternal(executor, tuple_element_size,
89 return EnqueueBuffersToInfeed(executor, buffers);
92 Status GpuTransferManager::TransferBufferToInfeed(se::StreamExecutor* executor,
96 TransferBufferToInfeedInternal(executor, size, source));
97 return EnqueueBuffersToInfeed(executor, {buffer});
101 se::StreamExecutor* executor, std::vector<gpu::InfeedBuffer*> buffers) {
103 se::Stream* stream = infeed_manager->GetStream(executor);
126 se::StreamExecutor* executor, int64 size, const void* source) {
136 se::Stream* stream = infeed_manager->GetStream(executor);
141 gpu::InfeedBuffer* buffer = new gpu::InfeedBuffer(executor, size);