Searched refs:flat_sharded_device_arrays (Results 1 – 1 of 1) sorted by relevance
334 py::list flat_sharded_device_arrays = in Call() local336 return cache_entry->out_pytree_def.Unflatten(flat_sharded_device_arrays); in Call()