diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-07 15:41:22 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-07 17:27:20 -0700 |
commit | fc7f0b296dd53d1b72af21d36d36b6bcc5291ea7 (patch) | |
tree | 46e76ead2391a3fb1232459189ad0b8d0d8066ac /tensorflow/contrib/lite/toco/tooling_util.cc | |
parent | 3a2f1cfb73fa6a21eba077485bdc08aa05646ad1 (diff) |
Add support for select (via tf.where) support to tflite.
PiperOrigin-RevId: 195734246
Diffstat (limited to 'tensorflow/contrib/lite/toco/tooling_util.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/tooling_util.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/tooling_util.cc b/tensorflow/contrib/lite/toco/tooling_util.cc index f82bb33535..1f56fe5c83 100644 --- a/tensorflow/contrib/lite/toco/tooling_util.cc +++ b/tensorflow/contrib/lite/toco/tooling_util.cc @@ -391,6 +391,7 @@ const char* OperatorTypeName(OperatorType type) { HANDLE_OPERATORTYPENAME_CASE(Exp) HANDLE_OPERATORTYPENAME_CASE(DynamicPartition) HANDLE_OPERATORTYPENAME_CASE(DynamicStitch) + HANDLE_OPERATORTYPENAME_CASE(Select) default: LOG(FATAL) << "Unhandled op type"; #undef HANDLE_OPERATORTYPENAME_CASE @@ -2097,6 +2098,8 @@ ArrayDataType ConvertIODataTypeToArrayDataType(IODataType type) { return ArrayDataType::kInt32; case INT64: return ArrayDataType::kInt64; + case BOOL: + return ArrayDataType::kBool; default: return ArrayDataType::kNone; } |