experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improves the ordering the Tensor data types.

Browse Source 
PiperOrigin-RevId: 248156212
This commit is contained in:
A. Unique TensorFlower 2019-05-14 09:59:30 -07:00 committed by TensorFlower Gardener
parent cd611387e4
commit 3c77cf763b
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tensorflow/lite/experimental/swift/Sources/Tensor.swift
View File

@ -58,10 +58,8 @@ public struct Tensor {
/// 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
/// Boolean.
case bool
/// 8-bit unsigned integer.
case uInt8
/// 16-bit signed integer.
@ -70,8 +68,10 @@ public enum TensorDataType: Equatable {
case int32
/// 64-bit signed integer.
case int64
/// Boolean.
case bool
/// 16-bit half precision floating point.
case float16
/// 32-bit single precision floating point.
case float32
/// Creates a new tensor data type from the given `TFL_Type` or `nil` if the data type is
/// unsupported or could not be determined because there was an error.
@ -79,10 +79,8 @@ public enum TensorDataType: Equatable {
/// - Parameter type: A data type supported by a tensor.
init?(type: TFL_Type) {
switch type {
case kTfLiteFloat32:
self = .float32
case kTfLiteFloat16:
self = .float16
case kTfLiteBool:
self = .bool
case kTfLiteUInt8:
self = .uInt8
case kTfLiteInt16:
@ -91,8 +89,10 @@ public enum TensorDataType: Equatable {
self = .int32
case kTfLiteInt64:
self = .int64
case kTfLiteBool:
self = .bool
case kTfLiteFloat16:
self = .float16
case kTfLiteFloat32:
self = .float32
case kTfLiteNoType:
fallthrough
default: