| 55 | } |
| 56 | |
| 57 | TF_DataType GetTensorFlowDataType(TfLiteType type) { |
| 58 | switch (type) { |
| 59 | case kTfLiteNoType: |
| 60 | return TF_FLOAT; |
| 61 | case kTfLiteFloat32: |
| 62 | return TF_FLOAT; |
| 63 | case kTfLiteFloat16: |
| 64 | return TF_HALF; |
| 65 | case kTfLiteInt16: |
| 66 | return TF_INT16; |
| 67 | case kTfLiteInt32: |
| 68 | return TF_INT32; |
| 69 | case kTfLiteUInt8: |
| 70 | return TF_UINT8; |
| 71 | case kTfLiteInt8: |
| 72 | return TF_INT8; |
| 73 | case kTfLiteInt64: |
| 74 | return TF_INT64; |
| 75 | case kTfLiteComplex64: |
| 76 | return TF_COMPLEX64; |
| 77 | case kTfLiteString: |
| 78 | return TF_STRING; |
| 79 | case kTfLiteBool: |
| 80 | return TF_BOOL; |
| 81 | } |
| 82 | } |
| 83 | |
| 84 | TfLiteType GetTensorFlowLiteType(TF_DataType type) { |
| 85 | switch (type) { |
no outgoing calls