Fix remove identity substitution when identity is model's output

Fix remove identity substitution when identity is model's output #699