diff --git a/core/lowering/passes/remove_bn_dim_check.cpp b/core/lowering/passes/remove_bn_dim_check.cpp index 6116522484..9b2a63b0a7 100644 --- a/core/lowering/passes/remove_bn_dim_check.cpp +++ b/core/lowering/passes/remove_bn_dim_check.cpp @@ -22,7 +22,7 @@ struct BNDimCheckRemoval { void run() { findBNDimCheckNodes(graph_->block()); torch::jit::EliminateDeadCode(graph_); - LOG_GRAPH("Post aten::addmm branch fusion: " << *graph_); + LOG_GRAPH("Post batch norm dim check removal: " << *graph_); } private: