| 110 | } |
| 111 | |
| 112 | ArrayDataType DataType::Deserialize(int tensor_type) { |
| 113 | switch (::tflite::TensorType(tensor_type)) { |
| 114 | case ::tflite::TensorType_FLOAT32: |
| 115 | return ArrayDataType::kFloat; |
| 116 | case ::tflite::TensorType_INT16: |
| 117 | return ArrayDataType::kInt16; |
| 118 | case ::tflite::TensorType_INT32: |
| 119 | return ArrayDataType::kInt32; |
| 120 | case ::tflite::TensorType_INT64: |
| 121 | return ArrayDataType::kInt64; |
| 122 | case ::tflite::TensorType_STRING: |
| 123 | return ArrayDataType::kString; |
| 124 | case ::tflite::TensorType_UINT8: |
| 125 | return ArrayDataType::kUint8; |
| 126 | case ::tflite::TensorType_BOOL: |
| 127 | return ArrayDataType::kBool; |
| 128 | case ::tflite::TensorType_COMPLEX64: |
| 129 | return ArrayDataType::kComplex64; |
| 130 | default: |
| 131 | LOG(FATAL) << "Unhandled tensor type '" << tensor_type << "'."; |
| 132 | } |
| 133 | } |
| 134 | |
| 135 | flatbuffers::Offset<flatbuffers::Vector<uint8_t>> DataBuffer::Serialize( |
| 136 | const Array& array, flatbuffers::FlatBufferBuilder* builder) { |