diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-15 14:49:01 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-15 14:52:56 -0800 |
commit | 41a8560e0c2fa3b8fa73622a15da53f5e1b8b6c8 (patch) | |
tree | 4b97df5ca9ecff906c675ded62a29f7cefd52339 /tensorflow/contrib/lite/kernels/test_util.cc | |
parent | 66f4f4cf31b86b7dd20f10ce6d968348b502f2ee (diff) |
Implement Split
PiperOrigin-RevId: 185904437
Diffstat (limited to 'tensorflow/contrib/lite/kernels/test_util.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/test_util.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/test_util.cc b/tensorflow/contrib/lite/kernels/test_util.cc index 6f56aa6bf3..373310bd87 100644 --- a/tensorflow/contrib/lite/kernels/test_util.cc +++ b/tensorflow/contrib/lite/kernels/test_util.cc @@ -187,6 +187,7 @@ void SingleOpModel::BuildInterpreter( for (const auto& shape : input_shapes) { int input_idx = interpreter_->inputs()[i++]; if (input_idx == kOptionalTensor) continue; + if (shape.empty()) continue; CHECK(interpreter_->ResizeInputTensor(input_idx, shape) == kTfLiteOk); } CHECK(interpreter_->AllocateTensors() == kTfLiteOk) |