Searched refs:output_shaped_buffer (Results 1 – 1 of 1) sorted by relevance
142 auto output_shaped_buffer = output_scoped_buffer.release(); in AllocateOutputTuple() local144 xla::Shape output_device_shape = output_shaped_buffer.on_device_shape(); in AllocateOutputTuple()148 stream, &output_shaped_buffer, &output_device_shape)); in AllocateOutputTuple()156 output_shaped_buffer, output_host_shape, output_device_shape, in AllocateOutputTuple()