build: fix typing/mypy issues
This commit is contained in:
1
tensorflow-stubs/keras/dtensor/__init__.pyi
Normal file
1
tensorflow-stubs/keras/dtensor/__init__.pyi
Normal file
@ -0,0 +1 @@
|
||||
from keras.api._v2.keras.dtensor import experimental as experimental
|
2
tensorflow-stubs/keras/dtensor/experimental/__init__.pyi
Normal file
2
tensorflow-stubs/keras/dtensor/experimental/__init__.pyi
Normal file
@ -0,0 +1,2 @@
|
||||
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
|
@ -0,0 +1 @@
|
||||
from keras.dtensor.optimizers import Adadelta as Adadelta, Adagrad as Adagrad, Adam as Adam, AdamW as AdamW, RMSprop as RMSprop, SGD as SGD
|
Reference in New Issue
Block a user