Searched refs:flat_device_arrays (Results 1 – 1 of 1) sorted by relevance
1187 py::list flat_device_arrays; in Call() local1193 flat_device_arrays.append(py::cast(std::move(outputs[i]))); in Call()1199 flat_device_arrays.append( in Call()1204 return cache_entry->out_pytree_def.Unflatten(flat_device_arrays); in Call()