Searched refs:compiled_module (Results 1 – 10 of 10) sorted by relevance
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_kernel_benchmark.py | 37 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 D | test_triton_wrapper.py | 15 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 D | layout_assignment_test.cc | 641 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 D | tf_to_tfrt.cc | 2021 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 D | fuse_module.py | 5 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 D | test_nnc_backend.cpp | 88 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 D | aot_model_compiler.cc | 135 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__.py | 7 compiled_module,
|
/aosp_15_r20/external/pytorch/torch/_functorch/ |
H A D | aot_autograd.py | 1557 compiled_module = aot_module variable
|
/aosp_15_r20/external/pytorch/test/functorch/ |
H A D | test_aotdispatch.py | 30 compiled_module, 3215 compiled_mod = compiled_module(mod, nop, nop) 3226 mod = compiled_module(nn.BatchNorm2d(4), nop, nop)
|