Home
last modified time | relevance | path

Searched defs:jit_module (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dcreate_dummy_torchscript_model.py27 jit_module = torch.jit.script(NeuralNetwork()) variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_to_jit_invocations.cc211 auto jit_module = tmp_module_builder.create<ModuleOp>(loc); in matchAndRewrite() local
/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dcollection.cpp462 auto jit_module = StealOrDefault<decltype(jit_modules_)>(jit_modules_); in materialize() local