diff options
Diffstat (limited to 'tensorflow/contrib/lite/kernels/test_util.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/test_util.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/kernels/test_util.cc b/tensorflow/contrib/lite/kernels/test_util.cc index 373310bd87..0bb28b50b2 100644 --- a/tensorflow/contrib/lite/kernels/test_util.cc +++ b/tensorflow/contrib/lite/kernels/test_util.cc @@ -141,8 +141,8 @@ void SingleOpModel::SetBuiltinOp(BuiltinOperator type, void SingleOpModel::SetCustomOp( const string& name, const std::vector<uint8_t>& custom_option, - const std::function<TfLiteRegistration*()>& registeration) { - custom_registrations_[name] = registeration; + const std::function<TfLiteRegistration*()>& registration) { + custom_registrations_[name] = registration; opcodes_.push_back( CreateOperatorCodeDirect(builder_, BuiltinOperator_CUSTOM, name.data())); operators_.push_back(CreateOperator( |