yt.frontends.flash.data_structures.FLASHGrid.partition_index_3d

FLASHGrid.partition_index_3d(ds, padding=0.0, rank_ratio=1)