| 88 | } |
| 89 | |
| 90 | Status GetNumPyType(const DataType& type, int* type_num) { |
| 91 | #define NUMPY_TYPE_CASE(ARROW_NAME, NPY_NAME) \ |
| 92 | case Type::ARROW_NAME: \ |
| 93 | *type_num = NPY_##NPY_NAME; \ |
| 94 | break; |
| 95 | |
| 96 | switch (type.id()) { |
| 97 | NUMPY_TYPE_CASE(UINT8, UINT8); |
| 98 | NUMPY_TYPE_CASE(INT8, INT8); |
| 99 | NUMPY_TYPE_CASE(UINT16, UINT16); |
| 100 | NUMPY_TYPE_CASE(INT16, INT16); |
| 101 | NUMPY_TYPE_CASE(UINT32, UINT32); |
| 102 | NUMPY_TYPE_CASE(INT32, INT32); |
| 103 | NUMPY_TYPE_CASE(UINT64, UINT64); |
| 104 | NUMPY_TYPE_CASE(INT64, INT64); |
| 105 | NUMPY_TYPE_CASE(HALF_FLOAT, FLOAT16); |
| 106 | NUMPY_TYPE_CASE(FLOAT, FLOAT32); |
| 107 | NUMPY_TYPE_CASE(DOUBLE, FLOAT64); |
| 108 | default: { |
| 109 | return Status::NotImplemented("Unsupported tensor type: ", type.ToString()); |
| 110 | } |
| 111 | } |
| 112 | #undef NUMPY_TYPE_CASE |
| 113 | |
| 114 | return Status::OK(); |
| 115 | } |
| 116 | |
| 117 | } // namespace |
| 118 |
no test coverage detected