github.com/alwaysproblem/mlserving-tutorial@v0.0.0-20221124033215-121cfddbfbf4/TFserving/CustomOp/custom-op/add_index/python/ops/add_index_ops.py (about)

     1  # Copyright 2018 The Sonnet 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  """Use add_index ops in python."""
    16  
    17  from __future__ import absolute_import
    18  from __future__ import division
    19  from __future__ import print_function
    20  
    21  from tensorflow.python.framework import load_library
    22  from tensorflow.python.platform import resource_loader
    23  
    24  add_index_ops = load_library.load_op_library(
    25      resource_loader.get_path_to_datafile('_add_index_ops.so')
    26  )
    27  add_index = add_index_ops.add_index