Converts a Variable NodeDef into a Constant NodeDef.
| 153 | |
| 154 | // Converts a Variable NodeDef into a Constant NodeDef. |
| 155 | void ConvertVariableToConstant(const NodeDef& variable_node, |
| 156 | const Tensor& variable_value, |
| 157 | NodeDef* const_node) { |
| 158 | const_node->set_name(variable_node.name()); |
| 159 | const_node->set_op("Const"); |
| 160 | (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype"); |
| 161 | variable_value.AsProtoTensorContent( |
| 162 | (*const_node->mutable_attr())["value"].mutable_tensor()); |
| 163 | } |
| 164 | |
| 165 | // Converts a ReadVariableOp NodeDef to an Identity NodeDef. |
| 166 | void ConvertReadVariableOpToIdentity(const NodeDef& node, |
no test coverage detected