Searched defs:config_modifier_hook (Results 1 – 3 of 3) sorted by relevance
71 const std::function<void(HloModuleConfig*)>& config_modifier_hook) { in LoadModuleFromData()122 const std::function<void(HloModuleConfig*)>& config_modifier_hook) { in LoadModuleFromFile()
35 const std::function<void(HloModuleConfig*)>& config_modifier_hook, in PrepareReferenceModule()
123 std::function<void(HloModuleConfig*)> config_modifier_hook) { in RunAndCompare()234 std::function<void(HloModuleConfig*)> config_modifier_hook) { in RunAndCompare()