Searched defs:script_module (Results 1 – 7 of 7) sorted by relevance
579 def init_fn(script_module): argument1013 def init_fn(script_module): argument1048 def init_fn(script_module): argument
658 def _finalize_scriptmodule(script_module): argument911 def init_fn(script_module): argument
43 ThroughputBenchmark::ThroughputBenchmark(const jit::Module& script_module) in ThroughputBenchmark()
87 def _script_module_reducer(self, script_module): argument
591 def script_module(*args, **kwargs): function
755 def script_module_to_buffer(script_module): argument
311 def all_backward_graphs(script_module, diff_graph_idx=None): argument320 def backward_graph(script_module, diff_graph_idx=None, skip_check=False): argument