ops-without-double.py

"""Print ops that have float but not double definitions."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from google3.net.proto2.python.public import text_format
from google3.pyglib import app
from google3.pyglib import flags
from google3.third_party.tensorflow.core.framework import op_def_pb2

import tensorflow as tf

FLAGS = flags.FLAGS
flags.DEFINE_string('ops', None, 'Path to ops file.')


def main(unused_argv):
  ops = op_def_pb2.OpList()
  text_format.Merge(open(FLAGS.ops).read(), ops)
  op_types = {}
  for op in ops.op:
    type_attrs = {}
    for attr in op.attr:
      if attr.type == 'type':
        type_attrs[attr.name] = list(attr.allowed_values.list.type)
    types = []
    for arg in list(op.input_arg) + list(op.output_arg):
      if arg.type_attr:
        types.extend(type_attrs[arg.type_attr])
      else:
        types.append(arg.type)
    op_types[op.name] = frozenset(types)
    #print('%s: %s' % (op.name, types))
  float32 = 1
  float64 = 2
  bad = []
  for op, types in op_types.items():
    if float32 in types and float64 not in types:
      #print('%s: %r' % (op, [tf.DType(t) for t in types]))
      bad.append(op)
  print('    ' + '\n    '.join(sorted(bad)))


if __name__ == '__main__':
  app.run()

I used --ops tensorflow/core/ops/compat/ops_history.v0.pbtxt

This can be made to work with the open source version by replacing the imports with

from google.protobuf import text_format
from tensorflow.python.platform import app, flags
from tensorflow.core.framework import op_def_pb2