update:
Add TensorConstant check in check_node_state method.