| 77 | } while (0) |
| 78 | |
| 79 | void MemToTensor(void* array_data_ptr, const size_t size, user_op::Tensor* tensor) { |
| 80 | switch (tensor->data_type()) { |
| 81 | case DataType::kFloat: TENSOR_MEM_ASSIGN(float); break; |
| 82 | case DataType::kDouble: TENSOR_MEM_ASSIGN(double); break; |
| 83 | case DataType::kBool: TENSOR_MEM_ASSIGN(bool); break; |
| 84 | case DataType::kInt8: TENSOR_MEM_ASSIGN(int8_t); break; |
| 85 | case DataType::kInt32: TENSOR_MEM_ASSIGN(int32_t); break; |
| 86 | case DataType::kInt64: TENSOR_MEM_ASSIGN(int64_t); break; |
| 87 | case DataType::kUInt8: TENSOR_MEM_ASSIGN(uint8_t); break; |
| 88 | case DataType::kFloat16: TENSOR_MEM_ASSIGN(float16); break; |
| 89 | default: |
| 90 | LOG(FATAL) << "OneFlow data type " << DataType_Name(tensor->data_type()) |
| 91 | << " is not supported yet."; |
| 92 | } |
| 93 | } |
| 94 | |
| 95 | void NumpyToTensor(PyObject* arg, user_op::Tensor* tensor) { |
| 96 | PyObject* ro_array = PyArray_FromAny(arg, nullptr, 0, 0, NPY_ARRAY_CARRAY_RO, nullptr); |
no test coverage detected