chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.arith
|
||||
---------
|
||||
.. automodule:: tvm.arith
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,123 @@
|
||||
.. 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.
|
||||
|
||||
tvm.contrib
|
||||
-----------
|
||||
.. automodule:: tvm.contrib
|
||||
|
||||
tvm.contrib.cblas
|
||||
~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.cblas
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.coreml_runtime
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.coreml_runtime
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.cublas
|
||||
~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.cublas
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.cublaslt
|
||||
~~~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.cublaslt
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.cudnn
|
||||
~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.cudnn
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.dlpack
|
||||
~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.dlpack
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.dnnl
|
||||
~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.dnnl
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.download
|
||||
~~~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.download
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.hipblas
|
||||
~~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.hipblas
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.mkl
|
||||
~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.mkl
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.nnpack
|
||||
~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.nnpack
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.pickle_memoize
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.pickle_memoize
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.random
|
||||
~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.random
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.thrust
|
||||
~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.thrust
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.tvmjs
|
||||
~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.tvmjs
|
||||
:members:
|
||||
|
||||
|
||||
tvm.contrib.cutlass
|
||||
~~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.cutlass
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
|
||||
tvm.contrib.hexagon
|
||||
~~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.contrib.hexagon
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.driver
|
||||
----------
|
||||
.. automodule:: tvm.driver
|
||||
|
||||
.. autofunction:: tvm.compile
|
||||
@@ -0,0 +1,23 @@
|
||||
.. 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.
|
||||
|
||||
tvm.error
|
||||
---------
|
||||
.. automodule:: tvm.error
|
||||
:members:
|
||||
:imported-members:
|
||||
:autosummary:
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.exec
|
||||
--------
|
||||
.. automodule:: tvm.exec
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,100 @@
|
||||
.. 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.
|
||||
|
||||
Python API
|
||||
==========
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
:caption: tvm
|
||||
|
||||
error
|
||||
ir
|
||||
arith
|
||||
instrument
|
||||
transform
|
||||
target
|
||||
driver
|
||||
testing
|
||||
exec
|
||||
support
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
:caption: tvm.runtime
|
||||
|
||||
runtime/runtime
|
||||
runtime/vm
|
||||
runtime/disco
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
:caption: tvm.relax
|
||||
|
||||
relax/relax
|
||||
relax/analysis
|
||||
relax/backend
|
||||
relax/block_builder
|
||||
relax/distributed
|
||||
relax/frontend
|
||||
relax/op
|
||||
relax/testing
|
||||
relax/training
|
||||
relax/transform
|
||||
relax/dpl
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
:caption: tvm.s_tir
|
||||
|
||||
s_tir/analysis
|
||||
s_tir/schedule
|
||||
s_tir/transform
|
||||
s_tir/dlight
|
||||
s_tir/backend
|
||||
s_tir/tensor_intrin
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
:caption: tvm.script
|
||||
|
||||
script/script
|
||||
script/parser
|
||||
script/ir_builder
|
||||
script/printer
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
:caption: tvm.te
|
||||
|
||||
te
|
||||
topi
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
:caption: tvm.s_tir.meta_schedule
|
||||
|
||||
meta_schedule
|
||||
|
||||
|
||||
|
||||
.. toctree::
|
||||
:maxdepth: 1
|
||||
:caption: Misc
|
||||
|
||||
rpc
|
||||
contrib
|
||||
@@ -0,0 +1,23 @@
|
||||
.. 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.
|
||||
|
||||
tvm.ir.instrument
|
||||
------------------
|
||||
.. automodule:: tvm.ir.instrument
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Path
|
||||
@@ -0,0 +1,24 @@
|
||||
.. 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.
|
||||
|
||||
tvm.ir
|
||||
------
|
||||
.. automodule:: tvm.ir
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
:autosummary:
|
||||
@@ -0,0 +1,146 @@
|
||||
.. 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.
|
||||
|
||||
tvm.s_tir.meta_schedule
|
||||
------------------------
|
||||
|
||||
tvm.s_tir.meta_schedule
|
||||
***********************
|
||||
.. automodule:: tvm.s_tir.meta_schedule
|
||||
:members:
|
||||
:imported-members:
|
||||
:autosummary:
|
||||
|
||||
tvm.s_tir.meta_schedule.arg_info
|
||||
********************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.arg_info
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.builder
|
||||
*******************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.builder
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.cost_model
|
||||
**********************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.cost_model
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.database
|
||||
********************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.database
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.feature_extractor
|
||||
*****************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.feature_extractor
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.measure_callback
|
||||
****************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.measure_callback
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.mutator
|
||||
*******************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.mutator
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.postproc
|
||||
********************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.postproc
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.runner
|
||||
******************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.runner
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.schedule_rule
|
||||
*************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.schedule_rule
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.search_strategy
|
||||
***************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.search_strategy
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.space_generator
|
||||
***************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.space_generator
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.task_scheduler
|
||||
**************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.task_scheduler
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.tir_integration
|
||||
***************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.tir_integration
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.relax_integration
|
||||
*****************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.relax_integration
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.trace_apply
|
||||
************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.trace_apply
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.schedule
|
||||
********************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.schedule
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.meta_schedule.post_optimization
|
||||
*****************************************
|
||||
.. automodule:: tvm.s_tir.meta_schedule.post_optimization
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.analysis
|
||||
------------------
|
||||
.. automodule:: tvm.relax.analysis
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,67 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.backend
|
||||
-----------------
|
||||
|
||||
tvm.relax.backend
|
||||
*****************
|
||||
.. automodule:: tvm.relax.backend
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.relax.backend.cuda
|
||||
**********************
|
||||
.. automodule:: tvm.relax.backend.cuda
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.backend.rocm
|
||||
**********************
|
||||
.. automodule:: tvm.relax.backend.rocm
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.backend.metal
|
||||
***********************
|
||||
.. automodule:: tvm.relax.backend.metal
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.backend.adreno
|
||||
************************
|
||||
.. automodule:: tvm.relax.backend.adreno
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.backend.gpu_generic
|
||||
*****************************
|
||||
.. automodule:: tvm.relax.backend.gpu_generic
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.backend.cpu_generic
|
||||
*****************************
|
||||
.. automodule:: tvm.relax.backend.cpu_generic
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.backend.contrib
|
||||
*************************
|
||||
.. automodule:: tvm.relax.backend.contrib
|
||||
:members:
|
||||
:no-index:
|
||||
@@ -0,0 +1,21 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.block_builder
|
||||
-----------------------
|
||||
.. automodule:: tvm.relax.block_builder
|
||||
:members:
|
||||
@@ -0,0 +1,28 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.distributed
|
||||
---------------------
|
||||
.. automodule:: tvm.relax.distributed
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.relax.distributed.transform
|
||||
*******************************
|
||||
.. automodule:: tvm.relax.distributed.transform
|
||||
:members:
|
||||
:no-index:
|
||||
@@ -0,0 +1,21 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.dpl
|
||||
-------------
|
||||
.. automodule:: tvm.relax.dpl
|
||||
:members:
|
||||
@@ -0,0 +1,60 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.frontend
|
||||
------------------
|
||||
.. automodule:: tvm.relax.frontend
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.relax.frontend.nn
|
||||
*********************
|
||||
.. automodule:: tvm.relax.frontend.nn
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: BlockBuilder
|
||||
:noindex:
|
||||
|
||||
tvm.relax.frontend.nn.llm
|
||||
*************************
|
||||
.. automodule:: tvm.relax.frontend.nn.llm
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.frontend.onnx
|
||||
***********************
|
||||
.. automodule:: tvm.relax.frontend.onnx
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.relax.frontend.stablehlo
|
||||
****************************
|
||||
.. automodule:: tvm.relax.frontend.stablehlo
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.relax.frontend.tflite
|
||||
*************************
|
||||
.. automodule:: tvm.relax.frontend.tflite
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.relax.frontend.torch
|
||||
************************
|
||||
.. automodule:: tvm.relax.frontend.torch
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,95 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.op
|
||||
------------
|
||||
|
||||
tvm.relax.op
|
||||
************
|
||||
.. automodule:: tvm.relax.op
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.nn
|
||||
***************
|
||||
.. automodule:: tvm.relax.op.nn
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.builtin
|
||||
********************
|
||||
.. automodule:: tvm.relax.op.builtin
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.ccl
|
||||
****************
|
||||
.. automodule:: tvm.relax.op.ccl
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.distributed
|
||||
************************
|
||||
.. automodule:: tvm.relax.op.distributed
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.grad
|
||||
*****************
|
||||
.. automodule:: tvm.relax.op.grad
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.image
|
||||
******************
|
||||
.. automodule:: tvm.relax.op.image
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.memory
|
||||
*******************
|
||||
.. automodule:: tvm.relax.op.memory
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.vision
|
||||
*******************
|
||||
.. automodule:: tvm.relax.op.vision
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.vm
|
||||
***************
|
||||
.. automodule:: tvm.relax.op.vm
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: Expr
|
||||
|
||||
tvm.relax.op.op_attrs
|
||||
*********************
|
||||
.. automodule:: tvm.relax.op.op_attrs
|
||||
:members:
|
||||
:exclude-members: Attrs
|
||||
@@ -0,0 +1,23 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax
|
||||
---------
|
||||
.. automodule:: tvm.relax
|
||||
:members:
|
||||
:imported-members:
|
||||
:exclude-members: BlockBuilder, Call, Span, GlobalVar, SourceName, TupleType, Type, FuncType
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.testing
|
||||
-----------------
|
||||
.. automodule:: tvm.relax.testing
|
||||
:members:
|
||||
:no-index:
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.relax.training
|
||||
------------------
|
||||
.. automodule:: tvm.relax.training
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,131 @@
|
||||
.. 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.
|
||||
|
||||
.. _api-relax-transformation:
|
||||
|
||||
tvm.relax.transform
|
||||
-------------------
|
||||
.. automodule:: tvm.relax.transform
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.relax.transform.legalize_ops
|
||||
*********************************
|
||||
.. automodule:: tvm.relax.transform.legalize_ops
|
||||
:members:
|
||||
|
||||
tvm.relax.transform.legalize_ops.binary
|
||||
========================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.binary
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.ccl
|
||||
=====================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.ccl
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.create
|
||||
========================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.create
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.datatype
|
||||
==========================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.datatype
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.distributed
|
||||
==============================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.distributed
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.grad
|
||||
======================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.grad
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.image
|
||||
=======================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.image
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.index
|
||||
=======================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.index
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.linear_algebra
|
||||
================================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.linear_algebra
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.manipulate
|
||||
=============================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.manipulate
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.nn
|
||||
====================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.nn
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.qdq
|
||||
=====================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.qdq
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.search
|
||||
========================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.search
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.statistical
|
||||
==============================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.statistical
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.unary
|
||||
=======================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.unary
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.vision
|
||||
========================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.vision
|
||||
:members:
|
||||
:no-index:
|
||||
|
||||
tvm.relax.transform.legalize_ops.adreno
|
||||
========================================
|
||||
.. automodule:: tvm.relax.transform.legalize_ops.adreno
|
||||
:members:
|
||||
:no-index:
|
||||
@@ -0,0 +1,23 @@
|
||||
.. 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.
|
||||
|
||||
tvm.rpc
|
||||
-------
|
||||
.. automodule:: tvm.rpc
|
||||
:members:
|
||||
:imported-members:
|
||||
:autosummary:
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.runtime.disco
|
||||
-----------------
|
||||
.. automodule:: tvm.runtime.disco
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,21 @@
|
||||
.. 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.
|
||||
|
||||
tvm.runtime
|
||||
-----------
|
||||
.. automodule:: tvm.runtime
|
||||
:members:
|
||||
@@ -0,0 +1,21 @@
|
||||
.. 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.
|
||||
|
||||
tvm.runtime.vm
|
||||
--------------------
|
||||
.. automodule:: tvm.runtime.vm
|
||||
:members:
|
||||
@@ -0,0 +1,21 @@
|
||||
.. 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.
|
||||
|
||||
tvm.s_tir.analysis
|
||||
------------------
|
||||
.. automodule:: tvm.s_tir.analysis
|
||||
:members:
|
||||
@@ -0,0 +1,42 @@
|
||||
.. 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.
|
||||
|
||||
tvm.s_tir.backend
|
||||
-----------------
|
||||
|
||||
tvm.s_tir.backend
|
||||
*****************
|
||||
.. automodule:: tvm.s_tir.backend
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.s_tir.backend.adreno
|
||||
************************
|
||||
.. automodule:: tvm.s_tir.backend.adreno
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.s_tir.backend.adreno.pipeline
|
||||
*********************************
|
||||
.. automodule:: tvm.s_tir.backend.adreno.pipeline
|
||||
:members:
|
||||
|
||||
tvm.s_tir.backend.adreno.transform
|
||||
***********************************
|
||||
.. automodule:: tvm.s_tir.backend.adreno.transform
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,66 @@
|
||||
.. 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.
|
||||
|
||||
tvm.s_tir.dlight
|
||||
----------------
|
||||
|
||||
tvm.s_tir.dlight
|
||||
****************
|
||||
.. automodule:: tvm.s_tir.dlight
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.s_tir.dlight.gpu
|
||||
********************
|
||||
.. automodule:: tvm.s_tir.dlight.gpu
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.dlight.adreno
|
||||
***********************
|
||||
.. automodule:: tvm.s_tir.dlight.adreno
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.dlight.cpu
|
||||
********************
|
||||
.. automodule:: tvm.s_tir.dlight.cpu
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.dlight.analysis
|
||||
*************************
|
||||
.. automodule:: tvm.s_tir.dlight.analysis
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.dlight.base
|
||||
*********************
|
||||
.. automodule:: tvm.s_tir.dlight.base
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.dlight.benchmark
|
||||
**************************
|
||||
.. automodule:: tvm.s_tir.dlight.benchmark
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.s_tir.schedule
|
||||
------------------
|
||||
.. automodule:: tvm.s_tir.schedule
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,73 @@
|
||||
.. 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.
|
||||
|
||||
tvm.s_tir.tensor_intrin
|
||||
-----------------------
|
||||
|
||||
tvm.s_tir.tensor_intrin
|
||||
***********************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.s_tir.tensor_intrin.cuda
|
||||
****************************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin.cuda
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.tensor_intrin.arm_cpu
|
||||
*******************************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin.arm_cpu
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.tensor_intrin.x86
|
||||
****************************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin.x86
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.tensor_intrin.rocm
|
||||
*****************************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin.rocm
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.tensor_intrin.metal
|
||||
*****************************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin.metal
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.tensor_intrin.hexagon
|
||||
*******************************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin.hexagon
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.tensor_intrin.riscv_cpu
|
||||
*********************************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin.riscv_cpu
|
||||
:members:
|
||||
:noindex:
|
||||
|
||||
tvm.s_tir.tensor_intrin.dot_product_common
|
||||
******************************************
|
||||
.. automodule:: tvm.s_tir.tensor_intrin.dot_product_common
|
||||
:members:
|
||||
:noindex:
|
||||
@@ -0,0 +1,23 @@
|
||||
.. 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.
|
||||
|
||||
tvm.s_tir.transform
|
||||
-------------------
|
||||
.. automodule:: tvm.s_tir.transform
|
||||
:members:
|
||||
:imported-members:
|
||||
:no-index:
|
||||
@@ -0,0 +1,46 @@
|
||||
.. 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.
|
||||
|
||||
tvm.script.ir_builder
|
||||
---------------------
|
||||
|
||||
tvm.script.ir_builder
|
||||
*********************
|
||||
.. automodule:: tvm.script.ir_builder
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.script.ir_builder.ir
|
||||
************************
|
||||
.. automodule:: tvm.script.ir_builder.ir
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.script.ir_builder.relax
|
||||
***************************
|
||||
.. automodule:: tvm.script.ir_builder.relax
|
||||
:members:
|
||||
|
||||
tvm.script.ir_builder.relax.distributed
|
||||
***************************************
|
||||
.. automodule:: tvm.script.ir_builder.relax.distributed
|
||||
:members:
|
||||
|
||||
tvm.script.ir_builder.tirx
|
||||
**************************
|
||||
.. automodule:: tvm.script.ir_builder.tirx
|
||||
:members:
|
||||
@@ -0,0 +1,37 @@
|
||||
.. 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.
|
||||
|
||||
tvm.script.parser
|
||||
-----------------
|
||||
|
||||
tvm.script.parser
|
||||
*****************
|
||||
.. automodule:: tvm.script.parser
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.script.parser.ir
|
||||
********************
|
||||
.. automodule:: tvm.script.parser.ir
|
||||
|
||||
tvm.script.parser.relax
|
||||
***********************
|
||||
.. automodule:: tvm.script.parser.relax
|
||||
|
||||
tvm.script.parser.tirx
|
||||
**********************
|
||||
.. automodule:: tvm.script.parser.tirx
|
||||
@@ -0,0 +1,25 @@
|
||||
.. 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.
|
||||
|
||||
tvm.script.printer
|
||||
------------------
|
||||
|
||||
tvm.script.printer
|
||||
******************
|
||||
.. automodule:: tvm.script.printer
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,35 @@
|
||||
.. 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.
|
||||
|
||||
tvm.script
|
||||
----------
|
||||
|
||||
tvm.script
|
||||
**********
|
||||
.. automodule:: tvm.script
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.script.relax
|
||||
****************
|
||||
.. automodule:: tvm.script.relax
|
||||
:members:
|
||||
|
||||
tvm.script.tirx
|
||||
***************
|
||||
.. automodule:: tvm.script.tirx
|
||||
:members:
|
||||
@@ -0,0 +1,73 @@
|
||||
.. 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.
|
||||
|
||||
tvm.support
|
||||
-----------
|
||||
.. automodule:: tvm.support
|
||||
:members:
|
||||
:imported-members:
|
||||
:autosummary:
|
||||
|
||||
tvm.support.cc
|
||||
~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.cc
|
||||
:members:
|
||||
|
||||
tvm.support.nvcc
|
||||
~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.nvcc
|
||||
:members:
|
||||
|
||||
tvm.support.rocm
|
||||
~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.rocm
|
||||
:members:
|
||||
|
||||
tvm.support.ndk
|
||||
~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.ndk
|
||||
:members:
|
||||
|
||||
tvm.support.xcode
|
||||
~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.xcode
|
||||
:members:
|
||||
|
||||
tvm.support.clang
|
||||
~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.clang
|
||||
:members:
|
||||
|
||||
tvm.support.emcc
|
||||
~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.emcc
|
||||
:members:
|
||||
|
||||
tvm.support.popen_pool
|
||||
~~~~~~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.popen_pool
|
||||
:members:
|
||||
|
||||
tvm.support.utils
|
||||
~~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.utils
|
||||
:members:
|
||||
|
||||
tvm.support.tar
|
||||
~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.support.tar
|
||||
:members:
|
||||
@@ -0,0 +1,29 @@
|
||||
.. 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.
|
||||
|
||||
tvm.target
|
||||
----------
|
||||
.. automodule:: tvm.target
|
||||
:members:
|
||||
:imported-members:
|
||||
:autosummary:
|
||||
|
||||
tvm.target.tag
|
||||
~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.target.tag
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,25 @@
|
||||
.. 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.
|
||||
|
||||
tvm.te
|
||||
------
|
||||
.. Exclude the ops imported from tirx.
|
||||
|
||||
.. automodule:: tvm.te
|
||||
:members:
|
||||
:imported-members:
|
||||
:autosummary:
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.testing
|
||||
-----------
|
||||
.. automodule:: tvm.testing
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,65 @@
|
||||
.. 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.
|
||||
|
||||
tvm.topi
|
||||
--------
|
||||
.. automodule:: tvm.topi
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
:autosummary:
|
||||
|
||||
tvm.topi.nn
|
||||
~~~~~~~~~~~
|
||||
|
||||
.. automodule:: tvm.topi.nn
|
||||
:members:
|
||||
:imported-members:
|
||||
:autosummary:
|
||||
|
||||
tvm.topi.image
|
||||
~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.topi.image
|
||||
:members:
|
||||
:imported-members:
|
||||
:autosummary:
|
||||
|
||||
tvm.topi.vision
|
||||
~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.topi.vision
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.topi.gpu
|
||||
~~~~~~~~~~~~
|
||||
.. automodule:: tvm.topi.gpu
|
||||
:members:
|
||||
:imported-members:
|
||||
:noindex:
|
||||
|
||||
tvm.topi.cpp
|
||||
~~~~~~~~~~~~
|
||||
.. automodule:: tvm.topi.cpp
|
||||
:members:
|
||||
:imported-members:
|
||||
|
||||
tvm.topi.testing
|
||||
~~~~~~~~~~~~~~~~
|
||||
.. automodule:: tvm.topi.testing
|
||||
:members:
|
||||
:imported-members:
|
||||
@@ -0,0 +1,22 @@
|
||||
.. 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.
|
||||
|
||||
tvm.ir.transform
|
||||
-----------------
|
||||
.. automodule:: tvm.ir.transform
|
||||
:members:
|
||||
:imported-members:
|
||||
Reference in New Issue
Block a user