diff options
author | 2018-09-26 18:26:51 -0700 | |
---|---|---|
committer | 2018-09-26 18:31:32 -0700 | |
commit | 51a6118e5bd85935b1d9ec0e68b92f1f98d14982 (patch) | |
tree | 5b3d7b469bebd510b678a2054c04311e4879fd09 /tensorflow/contrib/lite/testing | |
parent | 78ee51bb8c41d616c01674e405784a51e4ec98a5 (diff) |
Automated rollback of commit 82af048bc8c3c044c98a27b1c4c27bb62d4e4a14
PiperOrigin-RevId: 214705311
Diffstat (limited to 'tensorflow/contrib/lite/testing')
6 files changed, 10 insertions, 10 deletions
diff --git a/tensorflow/contrib/lite/testing/BUILD b/tensorflow/contrib/lite/testing/BUILD index f0bfec2338..55ef1172b2 100644 --- a/tensorflow/contrib/lite/testing/BUILD +++ b/tensorflow/contrib/lite/testing/BUILD @@ -164,7 +164,7 @@ cc_library( ":test_runner", "//tensorflow/contrib/lite:builtin_op_data", "//tensorflow/contrib/lite:framework", - "//tensorflow/contrib/lite/delegates/flex:delegate", + "//tensorflow/contrib/lite/delegates/eager:delegate", "//tensorflow/contrib/lite/kernels:builtin_ops", ], ) diff --git a/tensorflow/contrib/lite/testing/generate_examples.py b/tensorflow/contrib/lite/testing/generate_examples.py index 53bd88d087..014c80b5ef 100644 --- a/tensorflow/contrib/lite/testing/generate_examples.py +++ b/tensorflow/contrib/lite/testing/generate_examples.py @@ -343,7 +343,7 @@ def toco_convert(graph_def_str, input_tensors, output_tensors, opts = ("--input_arrays={0} --output_arrays={1}".format( ",".join(input_arrays), ",".join(output_tensors))) elif FLAGS.run_with_extended: - opts += " --allow_flex_ops --force_flex_ops" + opts += " --allow_eager_ops --force_eager_ops" cmd = ("%s --input_file=%s --output_file=%s %s > %s 2>&1" % (bin_path, graphdef_file.name, output_file.name, opts, stdout_file.name)) diff --git a/tensorflow/contrib/lite/testing/tflite_diff_flags.h b/tensorflow/contrib/lite/testing/tflite_diff_flags.h index ad889a2f19..3874bc31d7 100644 --- a/tensorflow/contrib/lite/testing/tflite_diff_flags.h +++ b/tensorflow/contrib/lite/testing/tflite_diff_flags.h @@ -57,7 +57,7 @@ DiffOptions ParseTfliteDiffFlags(int* argc, char** argv) { "[optional] Number of full runs in each pass."), tensorflow::Flag("delegate", &values.delegate, "[optional] Delegate to use for executing ops. Must be " - "`{\"\", FLEX}`"), + "`{\"\", EAGER}`"), }; bool no_inputs = *argc == 1; @@ -70,7 +70,7 @@ DiffOptions ParseTfliteDiffFlags(int* argc, char** argv) { values.input_layer_shape.empty() || values.output_layer.empty()) { fprintf(stderr, "%s", tensorflow::Flags::Usage(argv[0], flags).c_str()); return {}; - } else if (!(values.delegate == "" || values.delegate == "FLEX")) { + } else if (!(values.delegate == "" || values.delegate == "EAGER")) { fprintf(stderr, "%s", tensorflow::Flags::Usage(argv[0], flags).c_str()); return {}; } diff --git a/tensorflow/contrib/lite/testing/tflite_diff_util.h b/tensorflow/contrib/lite/testing/tflite_diff_util.h index 28b14bd143..f67992139f 100644 --- a/tensorflow/contrib/lite/testing/tflite_diff_util.h +++ b/tensorflow/contrib/lite/testing/tflite_diff_util.h @@ -45,7 +45,7 @@ struct DiffOptions { // second pass does multiple inferences back to back. int num_runs_per_pass; // Path to the delegate library to be loaded in order to execute ops. Must be - // `{"", FLEX}`. + // `{"", EAGER}`. string delegate; }; diff --git a/tensorflow/contrib/lite/testing/tflite_driver.cc b/tensorflow/contrib/lite/testing/tflite_driver.cc index 0a6da926be..1836eb53b9 100644 --- a/tensorflow/contrib/lite/testing/tflite_driver.cc +++ b/tensorflow/contrib/lite/testing/tflite_driver.cc @@ -17,7 +17,7 @@ limitations under the License. #include <iostream> #include "tensorflow/contrib/lite/builtin_op_data.h" -#include "tensorflow/contrib/lite/delegates/flex/delegate.h" +#include "tensorflow/contrib/lite/delegates/eager/delegate.h" #include "tensorflow/contrib/lite/testing/split.h" namespace tflite { @@ -138,8 +138,8 @@ class TfLiteDriver::Expectation { TfLiteDriver::TfLiteDriver(bool use_nnapi, const string& delegate_name) : use_nnapi_(use_nnapi) { - if (delegate_name == "FLEX") { - delegate_ = FlexDelegate::Create(); + if (delegate_name == "EAGER") { + delegate_ = EagerDelegate::Create(); } } diff --git a/tensorflow/contrib/lite/testing/tflite_driver.h b/tensorflow/contrib/lite/testing/tflite_driver.h index dc2a4e5877..aed35f877d 100644 --- a/tensorflow/contrib/lite/testing/tflite_driver.h +++ b/tensorflow/contrib/lite/testing/tflite_driver.h @@ -17,7 +17,7 @@ limitations under the License. #include <map> -#include "tensorflow/contrib/lite/delegates/flex/delegate.h" +#include "tensorflow/contrib/lite/delegates/eager/delegate.h" #include "tensorflow/contrib/lite/interpreter.h" #include "tensorflow/contrib/lite/kernels/register.h" #include "tensorflow/contrib/lite/model.h" @@ -53,7 +53,7 @@ class TfLiteDriver : public TestRunner { class Expectation; - std::unique_ptr<FlexDelegate> delegate_; + std::unique_ptr<EagerDelegate> delegate_; bool use_nnapi_ = false; std::unique_ptr<FlatBufferModel> model_; std::unique_ptr<Interpreter> interpreter_; |