blob: 3e7c98f4c62aab171e1f461d9dc8f41bb135623e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
%define %tensor_shape_typemap(TYPE1, TYPE2)
%typemap(in) (TYPE1, TYPE2) {
if (PyTuple_Check($input)) {
PyObject* seq = $input;
$1 = PySequence_Fast_GET_SIZE(seq);
$2 = (unsigned int*)PyMem_RawMalloc($1*sizeof(unsigned int));
if(!$2) {
PyErr_NoMemory();
SWIG_fail;
}
int size = (int)$1;
for(int i=0; i < size; i++) {
PyObject *longItem;
// Borrowed reference. No need to Py_DECREF
PyObject *item = PySequence_Fast_GET_ITEM(seq, i);
if(!item) {
PyErr_SetString(PyExc_TypeError, "Failed to read data from tuple");
SWIG_fail;
}
// New reference. Need to Py_DECREF
longItem = PyNumber_Long(item);
if(!longItem) {
Py_XDECREF(longItem);
PyErr_SetString(PyExc_TypeError, "All elements must be numbers");
SWIG_fail;
}
$2[i] = (unsigned int)PyLong_AsUnsignedLong(longItem);
Py_XDECREF(longItem);
}
} else {
PyErr_SetString(PyExc_TypeError, "Argument is not a tuple");
SWIG_fail;
}
}
%typemap(freearg) (TYPE1, TYPE2) {
PyMem_RawFree($2);
}
%enddef
%define %clear_tensor_shape_typemap(TYPE1, TYPE2)
%typemap(in) (TYPE1, TYPE2);
%typemap(freearg) (TYPE1, TYPE2);
%enddef
|
