Searched defs:fw_module (Results 1 – 2 of 2) sorted by relevance
251 def unlift_tokens(fw_module, fw_metadata, aot_config, bw_module=None): argument
523 fw_module, # Must be fw_module from aot_dispatch_*_graph argument