Searched defs:meta_module_fn (Results 1 – 1 of 1) sorted by relevance
153 def _test_simple_model_with_meta_device(self, meta_module_fn, init_fn=None): argument199 def meta_module_fn(): function208 def meta_module_fn(): function219 def meta_module_fn(): function230 def meta_module_fn(): function238 self, auto_wrap, meta_module_fn, init_fn=None argument292 def meta_module_fn(): function304 def meta_module_fn(): function319 def meta_module_fn(): function333 def meta_module_fn(): function[all …]