10 lines
212 B
Python
10 lines
212 B
Python
|
__author__ = "Alex Rogozhnikov"
|
||
|
|
|
||
|
|
from ..layers.tensorflow import Rearrange, Reduce, EinMix
|
||
|
|
|
||
|
|
keras_custom_objects = {
|
||
|
|
Rearrange.__name__: Rearrange,
|
||
|
|
Reduce.__name__: Reduce,
|
||
|
|
EinMix.__name__: EinMix,
|
||
|
|
}
|