1# Copyright 2015 The TensorFlow Authors. All Rights Reserved. 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================== 15"""ZeroOut ops Python library.""" 16import os.path 17 18import tensorflow as tf 19 20_zero_out_module = tf.load_op_library( 21 os.path.join(tf.compat.v1.resource_loader.get_data_files_path(), 22 'zero_out_op_kernel_2.so')) 23zero_out = _zero_out_module.zero_out 24zero_out2 = _zero_out_module.zero_out2 25zero_out3 = _zero_out_module.zero_out3 26