Searched defs:compiled_module (Results 1 – 6 of 6) sorted by relevance
641 std::unique_ptr<HloModule> compiled_module = in TEST_F() local927 auto compiled_module = in TEST_F() local959 auto compiled_module = in TEST_F() local992 auto compiled_module = in TEST_F() local1025 auto compiled_module = in TEST_F() local1049 auto compiled_module = in TEST_F() local
64 def verify_remove_inductor_deps(self, compiled_module): argument97 def check_bandwidth(self, compiled_module, num_gb): argument
135 auto compiled_module = torch::jit::detail::codegen_backend_module( in main() local
88 auto compiled_module = torch::jit::detail::codegen_backend_module( in TEST() local
1557 compiled_module = aot_module variable
2021 auto compiled_module = ModuleOp::create(loc, {"kernel"}); in CreateCompiledModule() local2134 CompiledModule compiled_module = in OutlineClusterOp() local