Searched defs:HloModule (Results 1 – 12 of 12) sorted by relevance
121 std::function<Status(const HloModule&, HloRunnerInterface*, HloModule*)> in RunAndCompare()232 std::function<Status(const HloModule&, HloRunnerInterface*, HloModule*)> in RunAndCompare()
36 const std::function<Status(const HloModule&, HloRunnerInterface*, in PrepareReferenceModule()
27 class HloModule; variable
31 class HloModule; variable
39 class HloModule; variable
57 HloModule::HloModule(const std::string& name, HloModuleConfig config) in HloModule() function in xla::HloModule971 HloModule::HloModule(const std::string& name, HloModuleConfig config, in HloModule() function in xla::HloModule
29 class HloModule; variable
28 class HloModule; variable
32 class HloModule; variable
47 class HloModule; variable
66 class HloModule; variable