Home
last modified time | relevance | path

Searched refs:TFModuleWrapper (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dmodule_wrapper_test.py37 wrapped_module = module_wrapper.TFModuleWrapper(module, 'test')
49 wrapped_module = module_wrapper.TFModuleWrapper(module, 'test')
72 wrapped_module = module_wrapper.TFModuleWrapper(
88 wrapped_module = module_wrapper.TFModuleWrapper(
105 wrapped_module = module_wrapper.TFModuleWrapper(
122 wrapped_module = module_wrapper.TFModuleWrapper(
135 wrapped_module = module_wrapper.TFModuleWrapper(
141 wrapped_module = module_wrapper.TFModuleWrapper(module, 'test')
147 self.assertFalse(module_wrapper.TFModuleWrapper.compat_v1_usage_recorded)
151 mock_tf_wrapped = module_wrapper.TFModuleWrapper(
[all …]
H A Dmodule_wrapper.py90 class TFModuleWrapper(FastModuleType): class
102 super(TFModuleWrapper, self).__init__(wrapped.__name__)
103 FastModuleType.set_getattr_callback(self, TFModuleWrapper._getattr)
105 TFModuleWrapper._getattribute)
162 not TFModuleWrapper.compat_v1_usage_recorded):
163 TFModuleWrapper.compat_v1_usage_recorded = True
255 super(TFModuleWrapper, self).__setattr__(arg, val)
270 super(TFModuleWrapper, self).__delattr__(name)
H A Ddeprecation_wrapper.py24 DeprecationWrapper = module_wrapper.TFModuleWrapper