Searched refs:compiled_module (Results 1 – 2 of 2) sorted by relevance
457 compiled_module, _ = compiler.ast_to_object(466 if key not in compiled_module.__dict__:467 compiled_module.__dict__[key] = val468 compiled = getattr(compiled_module, name)478 compiled_module.__dict__.keys())499 compiled_module.__dict__['ag_source_map__'])
681 std::unique_ptr<HloModule> compiled_module = in TEST_F() local690 ->RunBackend(std::move(compiled_module), in TEST_F()940 auto compiled_module = in TEST_F() local947 compiled_module->entry_computation()->root_instruction(); in TEST_F()972 auto compiled_module = in TEST_F() local979 compiled_module->entry_computation()->root_instruction(); in TEST_F()1005 auto compiled_module = in TEST_F() local1012 compiled_module->entry_computation()->root_instruction(); in TEST_F()1038 auto compiled_module = in TEST_F() local1045 compiled_module->entry_computation()->root_instruction(); in TEST_F()[all …]