|
|
@ -18,7 +18,8 @@ import contextlib
|
|
|
|
|
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
__all__ = [
|
|
|
|
'Constant', 'Uniform', 'Normal', 'Xavier', 'force_init_on_cpu',
|
|
|
|
'Constant', 'Uniform', 'Normal', 'Xavier', 'force_init_on_cpu',
|
|
|
|
'init_on_cpu', 'ConstantInitializer', 'UniformInitializer', 'NormalInitializer', 'XavierInitializer'
|
|
|
|
'init_on_cpu', 'ConstantInitializer', 'UniformInitializer',
|
|
|
|
|
|
|
|
'NormalInitializer', 'XavierInitializer'
|
|
|
|
]
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
_force_init_on_cpu_ = False
|
|
|
|
_force_init_on_cpu_ = False
|
|
|
|