toMap method

Map<String, dynamic> toMap()

Implementation

Map<String, dynamic> toMap() {
  Map<String, dynamic> map = {
    'shape': shape,
    'memoryFormat': memoryFormat.jniCode,
    'dtype': dtype.jniCode,
  };
  switch (dtype) {
    case DType.uint8:
      map['data'] = dataAsUin8List;
      break;
    case DType.int8:
      map['data'] = dataAsInt8List;
      break;
    case DType.int32:
      map['data'] = dataAsInt32List;
      break;
    case DType.float32:
      map['data'] = dataAsFloat32List;
      break;
    case DType.int64:
      map['data'] = dataAsInt64List;
      break;
    case DType.float64:
      map['data'] = dataAsFloat64List;
      break;
  }
  return map;
}