From 24df69d529826cf46fac3ba752345561f7b6fec9 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 18 May 2018 12:08:32 -0700 Subject: [PATCH] When converting an unsupported op in TOCO, we can possibly infer the output type from an attribute named Tout (this works, e.g., for RandomUniformInt). PiperOrigin-RevId: 197181552 --- tensorflow/contrib/lite/toco/import_tensorflow.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/contrib/lite/toco/import_tensorflow.cc b/tensorflow/contrib/lite/toco/import_tensorflow.cc index 3002857..9c35867 100644 --- a/tensorflow/contrib/lite/toco/import_tensorflow.cc +++ b/tensorflow/contrib/lite/toco/import_tensorflow.cc @@ -1362,6 +1362,9 @@ void ConvertUnsupportedOperator(const NodeDef& node, for (int i = 0; i < output_types.type_size(); ++i) { op->output_data_types.push_back(ConvertDataType(output_types.type(i))); } + } else if (HasAttr(node, "Tout")) { + const auto& output_type = GetDataTypeAttr(node, "Tout"); + op->output_data_types.push_back(ConvertDataType(output_type)); } } -- 2.7.4