| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| # pylint: disable=invalid-name, unused-argument |
| """FFI for tvm.node""" |
| import tvm._ffi |
| |
| # The implementations below are default ones when the corresponding |
| # functions are not available in the runtime only mode. |
| # They will be overriden via _init_api to the ones registered |
| # via TVM_REGISTER_GLOBAL in the compiler mode. |
| def AsRepr(obj): |
| return obj.type_key() + "(" + obj.handle.value + ")" |
| |
| |
| def NodeListAttrNames(obj): |
| return lambda x: 0 |
| |
| |
| def NodeGetAttr(obj, name): |
| raise AttributeError() |
| |
| |
| def SaveJSON(obj): |
| raise RuntimeError("Do not support object serialization in runtime only mode") |
| |
| |
| def LoadJSON(json_str): |
| raise RuntimeError("Do not support object serialization in runtime only mode") |
| |
| |
| # Exports functions registered via TVM_REGISTER_GLOBAL with the "node" prefix. |
| # e.g. TVM_REGISTER_GLOBAL("node.AsRepr") |
| tvm._ffi._init_api("node", __name__) |