Searched refs:nontuple_buffer (Results 1 – 1 of 1) sorted by relevance
278 ShapedBuffer nontuple_buffer = output.release(); in PopulateOutputs() local280 xla::ShapeUtil::MakeTupleShape({nontuple_buffer.on_host_shape()}), in PopulateOutputs()281 xla::ShapeUtil::MakeTupleShape({nontuple_buffer.on_device_shape()}), in PopulateOutputs()283 buffer.buffers().CopySubtreeFrom(nontuple_buffer.buffers(), in PopulateOutputs()