# 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=unused-argument """Internal module for quantization.""" from __future__ import absolute_import from tvm._ffi.function import _init_api _init_api("relay._quantize", __name__)
# 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. """The Relay virtual machine FFI namespace. """ from tvm._ffi.function import _init_api _init_api("relay._vm", __name__)
# pylint: disable=no-else-return, unidiomatic-typecheck, undefined-variable """The interface to the Module exposed from C++.""" from tvm._ffi.function import _init_api _init_api("relay._module", __name__)
# # 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. import numpy as np import tvm from tvm import relay from tvm._ffi.function import _init_api _init_api("tvm.relay.build_module") class BuildModule(object): def __init__(self): self.mod = relay.build_module._BuildModule() self._get_graph_json = self.mod["get_graph_json"] self._get_module = self.mod["get_module"] self._build = self.mod["build"] self._set_opt_level = self.mod["set_opt_level"] self._set_params_func = self.mod["set_params"] self._get_params_func = self.mod["get_params"] def build(self, func, target, target_host, params): tgts = [] for kv in target.items():
"""The interface to the Evaluator exposed from C++.""" from tvm._ffi.function import _init_api _init_api("relay._interpreter", __name__)
# pylint: disable=no-else-return, unidiomatic-typecheck, undefined-variable """The interface to the Environment exposed from C++.""" from tvm._ffi.function import _init_api _init_api("relay._env", __name__)
from __future__ import absolute_import from __future__ import division from __future__ import print_function from __future__ import unicode_literals import torch from tvm import relay # This registers all the schedules from ._torch_tvm import * from ._torch_tvm import _push_relay_expr from tvm._ffi.function import _init_api # This lets us use PackedFunc with torch_tvm _init_api("torch_tvm") def to_relay(pt_func, inputs): if type(pt_func) is not torch._C.Function: pt_func = torch.jit.trace(pt_func, inputs) handle = _push_relay_expr(pt_func.graph_for(*inputs), inputs) return _pop_relay_expr(handle)
# 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. """FFI exposing the Relay type inference and checking.""" from tvm._ffi.function import _init_api _init_api("relay._transform", __name__)
# # 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. import numpy as np import tvm from tvm import relay from tvm._ffi.function import _init_api _init_api("tvm.relay.build_module") class BuildModule(object): def __init__(self): self.mod = relay.build_module._BuildModule() self._get_graph_json = self.mod["get_graph_json"] self._get_module = self.mod["get_module"] self._build = self.mod["build"] self._set_opt_level = self.mod["set_opt_level"] self._set_params_func = self.mod["set_params"] self._get_params_func = self.mod["get_params"] def build(self, func, target, target_host, params): tgts = [] for kv in target.items():
# pylint: disable=no-else-return, unidiomatic-typecheck, undefined-variable """The interface of expr function exposed from C++.""" from tvm._ffi.function import _init_api _init_api("relay._expr", __name__)
# 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. """FFI exposing the passes for Relay program analysis.""" from tvm._ffi.function import _init_api _init_api("relay._analysis", __name__)
"""FFI exposing the Relay type inference and checking.""" from tvm._ffi.function import _init_api _init_api("relay._ir_pass", __name__)
# 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=no-else-return, unidiomatic-typecheck, undefined-variable """The interface of expr function exposed from C++.""" from tvm._ffi.function import _init_api _init_api("relay._base", __name__)
# 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. """The VM Object FFI namespace.""" from tvm._ffi.function import _init_api _init_api("_vmobj", __name__)
# 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=no-else-return, unidiomatic-typecheck, undefined-variable """The interface for building Relay functions exposed from C++.""" from tvm._ffi.function import _init_api _init_api("relay.build_module", __name__)
# 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. """FFI exposing the Relay type inference and checking.""" from tvm._ffi.function import _init_api _init_api("relay._ir_pass", __name__)