From 19cc7e35621b860b33208b11422c7ea28d15f21f Mon Sep 17 00:00:00 2001 From: xujiuxu Date: Mon, 27 Mar 2023 14:09:44 +0800 Subject: [PATCH] fixed 7883f4b from https://gitee.com/xujiuxu/tensorflow/pulls/2164 clean code --- tf_adapter/optimizers/frozen_variable_pass.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tf_adapter/optimizers/frozen_variable_pass.cc b/tf_adapter/optimizers/frozen_variable_pass.cc index 02582727e..951f66ab3 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"); -- Gitee