aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/toco/import_tensorflow.cc
diff options
context:
space:
mode:
authorGravatar Suharsh Sivakumar <suharshs@google.com>2018-09-13 15:34:43 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-13 15:39:27 -0700
commite59ddcca727340a8b45694a28cd9f52352607e63 (patch)
tree333e4da24dae2071c8946afd2c8a9def74f68889 /tensorflow/contrib/lite/toco/import_tensorflow.cc
parentea52ecd836098e0b1d37325cf1b91133f908547e (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.cc18
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);