robocar-training/tensorflow-stubs/keras/dtensor/experimental/__init__.pyi

3 lines
176 B
Python

from keras.api._v2.keras.dtensor.experimental import optimizers as optimizers
from keras.dtensor.layout_map import LayoutMap as LayoutMap, layout_map_scope as layout_map_scope