Searched refs:MakeOutputBuffers (Results 1 – 2 of 2) sorted by relevance
695 virtual std::vector<std::unique_ptr<PjRtBuffer>> MakeOutputBuffers(
1822 PjRtStreamExecutorExecutable::MakeOutputBuffers( in MakeOutputBuffers() function in xla::PjRtStreamExecutorExecutable1920 MakeOutputBuffers(device_ordinal, options, std::move(result_buffer), in ExecuteHelper()