| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/contrib/ |
| H A D | tedd.py | 259 tensor_intrin = None
266 if attr.tensor_intrin is not None:
267 tensor_intrin = str(attr.tensor_intrin.body)
269 tensor_intrin = tensor_intrin[0:-1] if tensor_intrin[-1] == "\n" else tensor_intrin
271 tensor_intrin = None
281 "intrin": tensor_intrin,
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/ |
| H A D | test_gemm_acc32_vnni.py | 23 from tvm.topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int32_cascadelake
24 from tvm.topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int32
|
| H A D | test_gemm_acc16.py | 21 from tvm.topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int16
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tests/python/contrib/ |
| H A D | test_gemm_acc32_vnni.py | 21 from topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int32_cascadelake
22 from topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int32
|
| H A D | test_gemm_acc16.py | 20 from topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int16
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/contrib/ |
| H A D | test_gemm_acc32_vnni.py | 21 from topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int32_cascadelake
22 from topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int32
|
| H A D | test_gemm_acc16.py | 20 from topi.x86.tensor_intrin import dot_16x1x16_uint8_int8_int16
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/te/ |
| H A D | schedule.py | 386 def tensorize(self, var, tensor_intrin): argument
397 _ffi_api.StageTensorize(self, var, tensor_intrin)
|
| H A D | __init__.py | 32 from .tensor_intrin import decl_tensor_intrin
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/cuda/ |
| H A D | conv2d_nhwc_tensorcore.py | 27 from .tensor_intrin import intrin_wmma_load_matrix_A
28 from .tensor_intrin import intrin_wmma_load_matrix_W
29 from .tensor_intrin import intrin_wmma_store_matrix
30 from .tensor_intrin import intrin_wmma_gemm
|
| H A D | conv2d_hwnc_tensorcore.py | 28 from .tensor_intrin import intrin_wmma_load_matrix_A
29 from .tensor_intrin import intrin_wmma_load_matrix_W
30 from .tensor_intrin import intrin_wmma_store_matrix
31 from .tensor_intrin import intrin_wmma_gemm
|
| H A D | conv3d_ndhwc_tensorcore.py | 27 from .tensor_intrin import intrin_wmma_load_matrix_A
28 from .tensor_intrin import intrin_wmma_load_matrix_W
29 from .tensor_intrin import intrin_wmma_store_matrix
30 from .tensor_intrin import intrin_wmma_gemm
|
| H A D | conv2d_nhwc_winograd.py | 28 from .tensor_intrin import intrin_wmma_load_matrix_A
29 from .tensor_intrin import intrin_wmma_load_matrix_W
30 from .tensor_intrin import intrin_wmma_store_matrix
31 from .tensor_intrin import intrin_wmma_gemm
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/ |
| H A D | __init__.py | 57 from .tensor_intrin import decl_tensor_intrin
|
| H A D | schedule.py | 538 def tensorize(self, var, tensor_intrin): argument
549 _api_internal._StageTensorize(self, var, tensor_intrin)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/ |
| H A D | __init__.py | 57 from .tensor_intrin import decl_tensor_intrin
|
| H A D | schedule.py | 538 def tensorize(self, var, tensor_intrin): argument
549 _api_internal._StageTensorize(self, var, tensor_intrin)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/ |
| H A D | schedule.h | 606 TensorIntrin tensor_intrin; variable
625 v->Visit("tensor_intrin", &tensor_intrin); in VisitAttrs()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/ |
| H A D | schedule.h | 606 TensorIntrin tensor_intrin; variable
625 v->Visit("tensor_intrin", &tensor_intrin); in VisitAttrs()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/arm_cpu/ |
| H A D | conv2d_int8.py | 28 from .tensor_intrin import dot_int8_int8_int32
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/arm_cpu/ |
| H A D | conv2d_int8.py | 28 from .tensor_intrin import dot_int8_int8_int32
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/te/ |
| H A D | schedule.h | 588 TensorIntrin tensor_intrin; variable
607 v->Visit("tensor_intrin", &tensor_intrin); in VisitAttrs()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/arm_cpu/ |
| H A D | conv2d_int8.py | 26 from .tensor_intrin import dot_int8_int8_int32
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/x86/ |
| H A D | conv2d_avx_common.py | 24 from .tensor_intrin import dot_16x1x16_uint8_int8_int32
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/recipe/gemm/ |
| H A D | gemm_int8.py | 23 from topi.cuda.tensor_intrin import dp4a
|