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