diff --git a/tf_adapter/optimizers/frozen_variable_pass.cc b/tf_adapter/optimizers/frozen_variable_pass.cc index 02582727ed20c7e0e26cb43c4f3558c2e6a7bc8a..951f66ab38c6142243f9a65f49951b45511ea635 100644 --- a/tf_adapter/optimizers/frozen_variable_pass.cc +++ b/tf_adapter/optimizers/frozen_variable_pass.cc @@ -49,10 +49,10 @@ class FrozenVariablePass : public GraphOptimizationPass { private: bool IsAllOutputsIdentity(const Node * const node) const; bool IsAllOutputsReadOp(const Node * const node) const; - bool IsNeedBuildPartitionedCall(const Node * const node); + bool IsNeedBuildPartitionedCall(const Node * const node) const; std::map GetGraphConfigs(const Graph &graph) const; void RemoveDeadNodes(Graph* g) const; - Status DoConstantFolding(const GraphOptimizationPassOptions &options, const uint64_t index); + Status DoConstantFolding(const GraphOptimizationPassOptions &options, const uint64_t index) const; }; struct StableNodeCompartor { @@ -79,7 +79,7 @@ bool FrozenVariablePass::IsAllOutputsReadOp(const Node * const node) const { return true; } -bool FrozenVariablePass::IsNeedBuildPartitionedCall(const Node * const node) { +bool FrozenVariablePass::IsNeedBuildPartitionedCall(const Node * const node) const { return ((node->type_string() == "Variable" || node->type_string() == "VariableV2") && IsAllOutputsIdentity(node)) || (node->type_string() == "VarHandleOp" && IsAllOutputsReadOp(node)); } @@ -105,7 +105,7 @@ void FrozenVariablePass::RemoveDeadNodes(Graph* g) const { } Status FrozenVariablePass::DoConstantFolding(const GraphOptimizationPassOptions &options, - const uint64_t index) { + const uint64_t index) const { ADP_LOG(INFO) << "Before do const folding " << options.session_options->config.DebugString(); if (options.device_set == nullptr) { return errors::Internal("Failed to get device set to run constant folding");