Searched refs:argument_buffer (Results 1 – 3 of 3) sorted by relevance
278 se::DeviceMemoryBase argument_buffer = owning->Release(); in CreateResultShapedBuffer() local279 *maybe_owning_memory = argument_buffer; in CreateResultShapedBuffer()280 result_buffer = argument_buffer; in CreateResultShapedBuffer()
239 ScopedShapedBuffer argument_buffer, in ExecuteReplicatedImpl()243 streams.back().get(), *argument, argument_buffer)); in ExecuteReplicatedImpl()244 argument_buffers.push_back(std::move(argument_buffer)); in ExecuteReplicatedImpl()
500 se::DeviceMemoryBase argument_buffer = owning->Release(); in ExecuteAsyncOnStreamImpl() local501 *maybe_owning_memory = argument_buffer; in ExecuteAsyncOnStreamImpl()502 result_buffer = argument_buffer; in ExecuteAsyncOnStreamImpl()