experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Disallow applying TFLite delegates by default for tests that rely on TfLiteDriver so that any testing with TFLite delegate has to be explicitly specified.

Browse Source 
PiperOrigin-RevId: 331101930
Change-Id: Ifee8177560fa46f5622ba237f23ae0fbcb702555
This commit is contained in:
Chao Mei 2020-09-11 00:56:40 -07:00 committed by TensorFlower Gardener
parent e743397337
commit 6da990b16e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/testing/tflite_driver.cc
View File

@ -361,7 +361,10 @@ TfLiteDriver::TfLiteDriver(DelegateType delegate_type, bool reference_kernel)
if (reference_kernel) {
resolver_.reset(new ops::builtin::BuiltinRefOpResolver);
} else {
resolver_.reset(new ops::builtin::BuiltinOpResolver);
// TODO(b/168278077): change back to use BuiltinOpResolver after zip tests
// are fully validated against TfLite delegates.
resolver_.reset(
new ops::builtin::BuiltinOpResolverWithoutDefaultDelegates());
ops::builtin::BuiltinOpResolver* buildinop_resolver_ =
reinterpret_cast<ops::builtin::BuiltinOpResolver*>(resolver_.get());
buildinop_resolver_->AddCustom("RFFT2D",