Home
last modified time | relevance | path

Searched refs:compiled_module (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_kernel_benchmark.py37 compiled_module = None
41 compiled_module is None, "Found multiple compiled modules"
43 compiled_module = v
45 self.assertTrue(compiled_module is not None)
46 return compiled_module
49 compiled_module = self.get_compiled_module()
64 def verify_remove_inductor_deps(self, compiled_module): argument
81 compiled_module.__file__, f"{compiled_module.__file__}.cleaned"
97 def check_bandwidth(self, compiled_module, num_gb): argument
157 compiled_module = self.get_compiled_module()
[all …]
H A Dtest_triton_wrapper.py15 compiled_module = None
19 compiled_module is None, "Found multiple compiled modules"
21 compiled_module = v
23 self.assertTrue(compiled_module is not None)
24 return compiled_module
41 compiled_module = self.get_compiled_module()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment_test.cc641 std::unique_ptr<HloModule> compiled_module = in TEST_F() local
650 ->RunBackend(std::move(compiled_module), in TEST_F()
927 auto compiled_module = in TEST_F() local
934 compiled_module->entry_computation()->root_instruction(); in TEST_F()
959 auto compiled_module = in TEST_F() local
966 compiled_module->entry_computation()->root_instruction(); in TEST_F()
992 auto compiled_module = in TEST_F() local
999 compiled_module->entry_computation()->root_instruction(); in TEST_F()
1025 auto compiled_module = in TEST_F() local
1032 compiled_module->entry_computation()->root_instruction(); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dtf_to_tfrt.cc2021 auto compiled_module = ModuleOp::create(loc, {"kernel"}); in CreateCompiledModule() local
2022 compiled_module->setAttr("tfrt.compiled", UnitAttr::get(ctx)); in CreateCompiledModule()
2023 compiled_module->setAttr( in CreateCompiledModule()
2027 SymbolTable compiled_module_symbol_table(compiled_module); in CreateCompiledModule()
2074 compiled_module.walk([](Operation *op) { op->removeAttr("_class"); }); in CreateCompiledModule()
2079 compiled_module.print(os); in CreateCompiledModule()
2086 compiled_module.erase(); // erase identical module in CreateCompiledModule()
2091 symbol_table->insert(compiled_module); in CreateCompiledModule()
2094 outlined_.insert({std::move(serialized), {compiled_module, compiled_func}}); in CreateCompiledModule()
2096 return {compiled_module, compiled_func, live_ins}; in CreateCompiledModule()
[all …]
/aosp_15_r20/external/pytorch/functorch/examples/compilation/
H A Dfuse_module.py5 from functorch.compile import compiled_module, tvm_compile
37 compiled_mod = compiled_module(mod, fw_compiler, bw_compiler)
/aosp_15_r20/external/pytorch/test/mobile/nnc/
H A Dtest_nnc_backend.cpp88 auto compiled_module = torch::jit::detail::codegen_backend_module( in TEST() local
93 compiled_module._save_for_mobile(ss); in TEST()
/aosp_15_r20/external/pytorch/binaries/
H A Daot_model_compiler.cc135 auto compiled_module = torch::jit::detail::codegen_backend_module( in main() local
137 compiled_module._save_for_mobile(output_model_name); in main()
/aosp_15_r20/external/pytorch/functorch/compile/
H A D__init__.py7 compiled_module,
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Daot_autograd.py1557 compiled_module = aot_module variable
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_aotdispatch.py30 compiled_module,
3215 compiled_mod = compiled_module(mod, nop, nop)
3226 mod = compiled_module(nn.BatchNorm2d(4), nop, nop)