Searched refs:BridgeLoggerConfig (Results 1 – 4 of 4) sorted by relevance
32 BridgeLoggerConfig::BridgeLoggerConfig(bool print_module_scope, in BridgeLoggerConfig() function in tensorflow::BridgeLoggerConfig45 inline static void Log(BridgeLoggerConfig::PrintCallbackFn print_callback, in Log()62 void BridgeLoggerConfig::printBeforeIfEnabled(mlir::Pass* pass, in printBeforeIfEnabled()68 void BridgeLoggerConfig::printAfterIfEnabled(mlir::Pass* pass, in printAfterIfEnabled()74 bool BridgeLoggerConfig::should_print(mlir::Pass* pass) { in should_print()
31 class BridgeLoggerConfig : public mlir::PassManager::IRPrinterConfig {33 explicit BridgeLoggerConfig(bool print_module_scope = false,
355 tf2xla.enableIRPrinting(std::make_unique<tensorflow::BridgeLoggerConfig>( in LegalizeToHlo()
35 pm->enableIRPrinting(std::make_unique<tensorflow::BridgeLoggerConfig>( in EnableLogging()