diff --git a/compiler/record-hessian/include/record-hessian/HessianObserver.h b/compiler/record-hessian/include/record-hessian/HessianObserver.h index 46a39b28b46..6bd9abb3102 100644 --- a/compiler/record-hessian/include/record-hessian/HessianObserver.h +++ b/compiler/record-hessian/include/record-hessian/HessianObserver.h @@ -19,10 +19,6 @@ #include "record-hessian/HessianComputer.h" -#include -#include -#include - namespace record_hessian { diff --git a/compiler/record-hessian/src/HessianObserver.cpp b/compiler/record-hessian/src/HessianObserver.cpp index 4740024dea5..be481969889 100644 --- a/compiler/record-hessian/src/HessianObserver.cpp +++ b/compiler/record-hessian/src/HessianObserver.cpp @@ -16,9 +16,6 @@ #include "record-hessian/HessianObserver.h" -#include -#include - using DataType = luci_interpreter::DataType; namespace record_hessian @@ -29,14 +26,15 @@ void HessianObserver::postTensorWrite(const luci::CircleNode *node, { auto node_outputs = loco::succs(node); - for (auto node : node_outputs) + for (auto node_output : node_outputs) { - auto _node = dynamic_cast(node); + auto cur_node = dynamic_cast(node_output); + assert(cur_node != nullptr); // TODO : ADD TCONV/DepthCONV cases - if (_node->opcode() == luci::CircleOpcode::FULLY_CONNECTED || - _node->opcode() == luci::CircleOpcode::CONV_2D) + if (cur_node->opcode() == luci::CircleOpcode::FULLY_CONNECTED || + cur_node->opcode() == luci::CircleOpcode::CONV_2D) { - _hessian_computer.recordHessian(_node, tensor); + _hessian_computer.recordHessian(cur_node, tensor); } } }