diff options
author | Suharsh Sivakumar <suharshs@google.com> | 2018-09-13 15:34:43 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-13 15:39:27 -0700 |
commit | e59ddcca727340a8b45694a28cd9f52352607e63 (patch) | |
tree | 333e4da24dae2071c8946afd2c8a9def74f68889 /tensorflow/contrib/lite/toco/import_tensorflow.cc | |
parent | ea52ecd836098e0b1d37325cf1b91133f908547e (diff) |
Automated rollback of commit 6b507a6de855a6f988100904229b7f46a5652b88
PiperOrigin-RevId: 212890622
Diffstat (limited to 'tensorflow/contrib/lite/toco/import_tensorflow.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/import_tensorflow.cc | 18 |
1 files changed, 0 insertions, 18 deletions
diff --git a/tensorflow/contrib/lite/toco/import_tensorflow.cc b/tensorflow/contrib/lite/toco/import_tensorflow.cc index eb36b3411d..9bc23c4b3c 100644 --- a/tensorflow/contrib/lite/toco/import_tensorflow.cc +++ b/tensorflow/contrib/lite/toco/import_tensorflow.cc @@ -58,7 +58,6 @@ using tensorflow::DT_STRING; using tensorflow::DT_UINT8; using tensorflow::GraphDef; using tensorflow::NodeDef; -using tensorflow::OpRegistry; using tensorflow::TensorProto; using tensorflow::TensorShapeProto; @@ -1080,23 +1079,6 @@ tensorflow::Status ConvertUnsupportedOperator( } else if (HasAttr(node, "Tout")) { const auto& output_type = GetDataTypeAttr(node, "Tout"); op->output_data_types.push_back(ConvertDataType(output_type)); - } else { - const tensorflow::OpDef* op_def = nullptr; - if (OpRegistry::Global()->LookUpOpDef(node.op(), &op_def).ok()) { - for (const auto& output_arg : op_def->output_arg()) { - if (HasAttr(node, output_arg.type_attr())) { - op->output_data_types.push_back( - ConvertDataType(GetDataTypeAttr(node, output_arg.type_attr()))); - } else { - LOG(INFO) << "Op node missing output type attribute: " << node.name(); - } - } - } - if (op->output_data_types.empty()) { - // TODO(b/113613439): Figure out how to propagate types for custom ops - // that have no OpDef. - LOG(INFO) << "Unable to determine output type for op: " << node.op(); - } } if (HasAttr(node, kAttrOutputShapes)) { const auto& output_shapes = GetListAttr(node, kAttrOutputShapes); |