message: Fix the check used to determine training quantization ops.
diff: @@ -376,7 +376,7 @@ class QuantizationMode(object):
  })
  for node_def in self._graph_def.node:
  - if any(op in node_def.name for op in training_quant_ops):
  + if node_def.op in training_quant_ops:
  return True
  return False
  