TensorFlow ops registration