load_op.py 156 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 + '/../c++/build/op_lib.so')