load_op.py 149 B

12345
  1. import os
  2. import tensorflow as tf
  3. dir_path = os.path.dirname(os.path.realpath(__file__))
  4. op_lib = tf.load_op_library(dir_path + '/build/op_lib.so')