2
0

tf-operator-json.py 476 B

1234567891011121314
  1. #!/usr/bin/env python
  2. from tensorflow.core.framework import op_def_pb2
  3. from google.protobuf import text_format
  4. input_file = '../third_party/tensorflow/tensorflow/core/ops/ops.pbtxt';
  5. output_file = '../src/tf-operator.pb'
  6. with open(input_file) as input_handle:
  7. ops_list = op_def_pb2.OpList()
  8. text_format.Merge(input_handle.read(), ops_list)
  9. data = ops_list.SerializeToString()
  10. with open(output_file, 'wb') as output_handle:
  11. output_handle.write(data)