Searched defs:jit_module (Results 1 – 3 of 3) sorted by relevance
27 jit_module = torch.jit.script(NeuralNetwork()) variable
211 auto jit_module = tmp_module_builder.create<ModuleOp>(loc); in matchAndRewrite() local
462 auto jit_module = StealOrDefault<decltype(jit_modules_)>(jit_modules_); in materialize() local