| /dports/misc/tvm/incubator-tvm-0.6.1/src/runtime/contrib/cblas/ |
| H A D | gemm_common.h | 34 inline int ColumnStride(DLTensor *tensor) { in ColumnStride()
45 inline int ElementStride(DLTensor *tensor) { in ElementStride()
54 inline bool IsInPlaceTransposed(DLTensor *tensor) { in IsInPlaceTransposed()
70 DLTensor *A = args[0]; in CallGemm()
71 DLTensor *B = args[1]; in CallGemm()
72 DLTensor *C = args[2]; in CallGemm()
108 inline int ColumnStride3D(DLTensor *tensor) { in ColumnStride3D()
118 inline int ElementStride3D(DLTensor *tensor) { in ElementStride3D()
139 DLTensor *A = args[0]; in CallBatchGemm()
140 DLTensor *B = args[1]; in CallBatchGemm()
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/runtime/contrib/cblas/ |
| H A D | gemm_common.h | 34 inline int ColumnStride(DLTensor *tensor) { in ColumnStride()
45 inline int ElementStride(DLTensor *tensor) { in ElementStride()
54 inline bool IsInPlaceTransposed(DLTensor *tensor) { in IsInPlaceTransposed()
70 DLTensor *A = args[0]; in CallGemm()
71 DLTensor *B = args[1]; in CallGemm()
72 DLTensor *C = args[2]; in CallGemm()
108 inline int ColumnStride3D(DLTensor *tensor) { in ColumnStride3D()
118 inline int ElementStride3D(DLTensor *tensor) { in ElementStride3D()
139 DLTensor *A = args[0]; in CallBatchGemm()
140 DLTensor *B = args[1]; in CallBatchGemm()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/cblas/ |
| H A D | gemm_common.h | 36 inline int ColumnStride(DLTensor* tensor) { in ColumnStride()
47 inline int ElementStride(DLTensor* tensor) { in ElementStride()
68 DLTensor* A = args[0]; in CallGemm()
69 DLTensor* B = args[1]; in CallGemm()
70 DLTensor* C = args[2]; in CallGemm()
107 DLTensor* A = args[0]; in CallU8S8S32Gemm()
108 DLTensor* B = args[1]; in CallU8S8S32Gemm()
109 DLTensor* C = args[2]; in CallU8S8S32Gemm()
177 DLTensor* A = args[0]; in CallBatchGemm()
178 DLTensor* B = args[1]; in CallBatchGemm()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/runtime/contrib/example_ext_runtime/ |
| H A D | example_ext_runtime.cc | 76 DLTensor* arg0 = static_cast<DLTensor*>(value[0].v_handle); in Add()
77 DLTensor* arg1 = static_cast<DLTensor*>(value[1].v_handle); in Add()
78 DLTensor* out = static_cast<DLTensor*>(value[2].v_handle); in Add()
93 DLTensor* arg0 = static_cast<DLTensor*>(value[0].v_handle); in Sub()
94 DLTensor* arg1 = static_cast<DLTensor*>(value[1].v_handle); in Sub()
95 DLTensor* out = static_cast<DLTensor*>(value[2].v_handle); in Sub()
110 DLTensor* arg0 = static_cast<DLTensor*>(value[0].v_handle); in Mul()
111 DLTensor* arg1 = static_cast<DLTensor*>(value[1].v_handle); in Mul()
112 DLTensor* out = static_cast<DLTensor*>(value[2].v_handle); in Mul()
149 DLTensor* arg = args[i]; in GetFunction()
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/runtime/contrib/example_ext_runtime/ |
| H A D | example_ext_runtime.cc | 76 DLTensor* arg0 = static_cast<DLTensor*>(value[0].v_handle); in Add()
77 DLTensor* arg1 = static_cast<DLTensor*>(value[1].v_handle); in Add()
78 DLTensor* out = static_cast<DLTensor*>(value[2].v_handle); in Add()
93 DLTensor* arg0 = static_cast<DLTensor*>(value[0].v_handle); in Sub()
94 DLTensor* arg1 = static_cast<DLTensor*>(value[1].v_handle); in Sub()
95 DLTensor* out = static_cast<DLTensor*>(value[2].v_handle); in Sub()
110 DLTensor* arg0 = static_cast<DLTensor*>(value[0].v_handle); in Mul()
111 DLTensor* arg1 = static_cast<DLTensor*>(value[1].v_handle); in Mul()
112 DLTensor* out = static_cast<DLTensor*>(value[2].v_handle); in Mul()
149 DLTensor* arg = args[i]; in GetFunction()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/thrust/ |
| H A D | thrust.cu | 40 void thrust_sort(DLTensor* input, in thrust_sort()
41 DLTensor* out_values, in thrust_sort()
42 DLTensor* out_indices, in thrust_sort()
71 void thrust_sort_common(DLTensor* input, in thrust_sort_common()
134 DLTensor* input = args[0]; in __anonbd1c66130102()
135 DLTensor* valid_count = args[1]; in __anonbd1c66130102()
136 DLTensor* values_out = args[2]; in __anonbd1c66130102()
137 DLTensor* indices_out = args[3]; in __anonbd1c66130102()
153 DLTensor* input = args[0]; in __anonbd1c66130302()
154 DLTensor* values_out = args[1]; in __anonbd1c66130302()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/runtime/ |
| H A D | ndarray.h | 114 inline const DLTensor* operator->() const;
121 inline void CopyFrom(DLTensor* other);
129 inline void CopyTo(DLTensor* other) const;
191 DLTensor* from, DLTensor* to, TVMStreamHandle stream = nullptr);
253 DLTensor dl_tensor;
363 inline size_t GetDataSize(const DLTensor& arr) {
372 inline void NDArray::CopyFrom(DLTensor* other) {
383 inline void NDArray::CopyTo(DLTensor* other) const {
396 const DLTensor* dptr = operator->();
408 inline const DLTensor* NDArray::operator->() const {
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/runtime/ |
| H A D | ndarray.h | 114 inline const DLTensor* operator->() const;
121 inline void CopyFrom(DLTensor* other);
129 inline void CopyTo(DLTensor* other) const;
191 DLTensor* from, DLTensor* to, TVMStreamHandle stream = nullptr);
253 DLTensor dl_tensor;
363 inline size_t GetDataSize(const DLTensor& arr) {
372 inline void NDArray::CopyFrom(DLTensor* other) {
383 inline void NDArray::CopyTo(DLTensor* other) const {
396 const DLTensor* dptr = operator->();
408 inline const DLTensor* NDArray::operator->() const {
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/runtime/ |
| H A D | ndarray.h | 67 inline const DLTensor* operator->() const;
76 inline void CopyFrom(const DLTensor* other);
92 inline void CopyTo(DLTensor* other) const;
158 TVM_DLL static void CopyFromTo(const DLTensor* from, DLTensor* to,
204 inline bool SaveDLTensor(dmlc::Stream* strm, const DLTensor* tensor);
221 DLTensor dl_tensor;
298 inline size_t GetDataSize(const DLTensor& arr) { in GetDataSize()
312 inline bool IsContiguous(const DLTensor& arr) { in IsContiguous()
327 inline void NDArray::CopyFrom(const DLTensor* other) { in CopyFrom()
338 inline void NDArray::CopyTo(DLTensor* other) const { in CopyTo()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/arm_compute_lib/ |
| H A D | acl_utils.h | 66 const DLTensor* scale = nullptr,
67 const DLTensor* offset = nullptr);
80 const DLDataType& dtype, const DLTensor* scale = nullptr,
81 const DLTensor* offset = nullptr);
120 std::vector<T> GetVectorFromDLTensor(const DLTensor* tensor);
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/rust/tvm-graph-rt/tests/test_tvm_dso/src/ |
| H A D | main.rs | 21 use tvm_graph_rt::{DLTensor, DsoModule, Module};
33 let mut a_dl: DLTensor = (&mut a).into(); in main()
34 let mut b_dl: DLTensor = (&mut b).into(); in main()
35 let mut c_dl: DLTensor = (&mut c).into(); in main()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/rust/tvm-graph-rt/tests/test_wasm32/src/ |
| H A D | main.rs | 36 use tvm_graph_rt::{DLTensor, Module as _, SystemLibModule};
44 let mut a_dl: DLTensor = (&mut a).into(); in main()
45 let mut b_dl: DLTensor = (&mut b).into(); in main()
46 let mut c_dl: DLTensor = (&mut c).into(); in main()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/rust/runtime/tests/test_tvm_dso/src/ |
| H A D | main.rs | 25 use tvm_runtime::{DLTensor, DsoModule, Module};
37 let mut a_dl: DLTensor = (&mut a).into(); in main()
38 let mut b_dl: DLTensor = (&mut b).into(); in main()
39 let mut c_dl: DLTensor = (&mut c).into(); in main()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/rust/runtime/tests/test_tvm_dso/src/ |
| H A D | main.rs | 25 use tvm_runtime::{DLTensor, DsoModule, Module};
37 let mut a_dl: DLTensor = (&mut a).into(); in main()
38 let mut b_dl: DLTensor = (&mut b).into(); in main()
39 let mut c_dl: DLTensor = (&mut c).into(); in main()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/ethosn/ |
| H A D | ethosn_device.cc | 76 void CopyOutput(dl::Buffer* source_buffers[], std::vector<DLTensor*>* outputs) { in CopyOutput()
77 for (DLTensor* tensor : *outputs) { in CopyOutput()
88 const std::vector<DLTensor*>& tensors) { in CreateBuffers()
103 std::vector<DLTensor*> inputs(input_order.size()); in Inference()
108 std::vector<DLTensor*> outputs(output_order.size()); in Inference()
142 for (DLTensor* tensor : outputs) { in Inference()
191 const DLTensor* tensor = args[argc]; in __anond7836f1e0102()
202 std::vector<DLTensor*> outputs; in Inference()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/rust/tvm-graph-rt/tests/test_tvm_basic/src/ |
| H A D | main.rs | 21 use tvm_graph_rt::{DLTensor, Module as _, SystemLibModule};
33 let mut a_dl: DLTensor = (&mut a).into(); in main()
34 let mut b_dl: DLTensor = (&mut b).into(); in main()
35 let mut c_dl: DLTensor = (&mut c).into(); in main()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/rust/runtime/tests/test_tvm_basic/src/ |
| H A D | main.rs | 25 use tvm_runtime::{DLTensor, Module as _, SystemLibModule};
37 let mut a_dl: DLTensor = (&mut a).into(); in main()
38 let mut b_dl: DLTensor = (&mut b).into(); in main()
39 let mut c_dl: DLTensor = (&mut c).into(); in main()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/rust/runtime/tests/test_tvm_basic/src/ |
| H A D | main.rs | 25 use tvm_runtime::{DLTensor, Module as _, SystemLibModule};
37 let mut a_dl: DLTensor = (&mut a).into(); in main()
38 let mut b_dl: DLTensor = (&mut b).into(); in main()
39 let mut c_dl: DLTensor = (&mut c).into(); in main()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/graph/ |
| H A D | graph_runtime.cc | 44 inline size_t GetDataAlignment(const DLTensor& arr) { in GetDataAlignment()
101 void GraphRuntime::SetInput(int index, DLTensor* data_in) { in SetInput()
114 const DLTensor* old_t = data_entry_[eid].operator->(); in SetInputZeroCopy()
127 for (DLTensor* t : input_dltensors_[eid]) { in SetInputZeroCopy()
170 void GraphRuntime::CopyOutputTo(int index, DLTensor* data_out) { in CopyOutputTo()
241 const DLTensor* tmp = data_entry_[eid].operator->(); in ShareParams()
307 const DLTensor* tmp = data_entry_[i].operator->(); in SetupStorage()
325 std::vector<DLTensor> args; in SetupOpExecs()
359 DLTensor* t = &arg_ptr->args[i]; in CreateTVMOp()
377 DLTensor* from = static_cast<DLTensor*>(arg_ptr->arg_values[0].v_handle); in CreateTVMOp()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/runtime/graph/ |
| H A D | graph_runtime.cc | 44 inline size_t GetDataAlignment(const DLTensor& arr) { in GetDataAlignment()
106 void GraphRuntime::SetInput(int index, DLTensor* data_in) { in SetInput()
119 const DLTensor* old_t = data_entry_[eid].operator->(); in SetInputZeroCopy()
132 for (DLTensor* t : input_dltensors_[eid]) { in SetInputZeroCopy()
246 const DLTensor* tmp = data_entry_[eid].operator->(); in ShareParams()
316 const DLTensor* tmp = data_entry_[i].operator->(); in SetupStorage()
334 std::vector<DLTensor> args; in SetupOpExecs()
362 const std::vector<DLTensor>& args, in CreateTVMOp()
372 DLTensor* t = &arg_ptr->args[i]; in CreateTVMOp()
390 DLTensor* from = static_cast<DLTensor*>(arg_ptr->arg_values[0].v_handle); in CreateTVMOp()
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/runtime/graph/ |
| H A D | graph_runtime.cc | 44 inline size_t GetDataAlignment(const DLTensor& arr) { in GetDataAlignment()
106 void GraphRuntime::SetInput(int index, DLTensor* data_in) { in SetInput()
119 const DLTensor* old_t = data_entry_[eid].operator->(); in SetInputZeroCopy()
132 for (DLTensor* t : input_dltensors_[eid]) { in SetInputZeroCopy()
246 const DLTensor* tmp = data_entry_[eid].operator->(); in ShareParams()
316 const DLTensor* tmp = data_entry_[i].operator->(); in SetupStorage()
334 std::vector<DLTensor> args; in SetupOpExecs()
362 const std::vector<DLTensor>& args, in CreateTVMOp()
372 DLTensor* t = &arg_ptr->args[i]; in CreateTVMOp()
390 DLTensor* from = static_cast<DLTensor*>(arg_ptr->arg_values[0].v_handle); in CreateTVMOp()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/sort/ |
| H A D | sort.cc | 53 DLTensor* input = args[0]; in __anonb5140abd0102()
54 DLTensor* sort_num = args[1]; in __anonb5140abd0102()
55 DLTensor* output = args[2]; in __anonb5140abd0102()
128 void argsort(DLTensor* input, DLTensor* output, int32_t axis, bool is_ascend) { in argsort()
171 DLTensor* input = args[0]; in __anonb5140abd0202()
172 DLTensor* output = args[1]; in __anonb5140abd0202()
247 void topk(DLTensor* input, DLTensor* out_values, DLTensor* out_indices, int k, int axis, in topk()
305 DLTensor* input = args[0]; in __anonb5140abd0302()
306 DLTensor* values_out = nullptr; in __anonb5140abd0302()
307 DLTensor* indices_out = nullptr; in __anonb5140abd0302()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/runtime/contrib/sort/ |
| H A D | sort.cc | 57 DLTensor *input = args[0]; in __anone763a8020102()
58 DLTensor *sort_num = args[1]; in __anone763a8020102()
59 DLTensor *output = args[2]; in __anone763a8020102()
131 void argsort(DLTensor* input, DLTensor* output, int32_t axis, bool is_ascend) { in argsort()
175 DLTensor *input = args[0]; in __anone763a8020202()
176 DLTensor *output = args[1]; in __anone763a8020202()
250 void topk(DLTensor* input, in topk()
251 DLTensor* out_values, in topk()
252 DLTensor* out_indices, in topk()
314 DLTensor* input = args[0]; in __anone763a8020302()
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/runtime/contrib/sort/ |
| H A D | sort.cc | 57 DLTensor *input = args[0]; in __anon9f6e80180102()
58 DLTensor *sort_num = args[1]; in __anon9f6e80180102()
59 DLTensor *output = args[2]; in __anon9f6e80180102()
131 void argsort(DLTensor* input, DLTensor* output, int32_t axis, bool is_ascend) { in argsort()
175 DLTensor *input = args[0]; in __anon9f6e80180202()
176 DLTensor *output = args[1]; in __anon9f6e80180202()
250 void topk(DLTensor* input, in topk()
251 DLTensor* out_values, in topk()
252 DLTensor* out_indices, in topk()
314 DLTensor* input = args[0]; in __anon9f6e80180302()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/runtime/micro/standalone/ |
| H A D | utvm_graph_runtime.h | 101 void CopyFrom(DLTensor* src);
103 void CopyTo(DLTensor* dst) const;
105 DLTensor ToDLTensor();
128 void SetInput(int index, DLTensor* data_in);
130 void CopyOutputTo(int index, DLTensor* data_out);
|