       self = .int32
     case kTfLiteInt64:
       self = .int64
-    case kTfLiteBool:
-      self = .bool
+    case kTfLiteFloat16:
+      self = .float16
+    case kTfLiteFloat32:
+      self = .float32
     case kTfLiteNoType:
       fallthrough
     default:


/// Supported TensorFlow Lite tensor data types.
public enum TensorDataType: Equatable {
  /// 32-bit single precision floating point.
  case float32
  /// 16-bit half precision floating point.
  case float16
  /// 8-bit unsigned integer.
  case uInt8
  /// 16-bit signed integer.

  /// - Parameter type: A data type supported by a tensor.
  init?(type: TFL_Type) {
    switch type {
    case kTfLiteBool:
      self = .bool
    case kTfLiteUInt8:
      self = .uInt8
    case kTfLiteInt16:
