Fixed reference names.

PiperOrigin-RevId: 304138406
Change-Id: Id3d9fcdcdee2ae285e640f2a870a900d93a7f587
This commit is contained in:
Chao Mei 2020-04-01 01:43:43 -07:00 committed by TensorFlower Gardener
parent bda5ba03da
commit 031804922d
2 changed files with 2 additions and 2 deletions

View File

@ -108,7 +108,7 @@ TFLITE_ATTRIBUTE_WEAK Interpreter::TfLiteDelegatePtr AcquireFlexDelegate() {
} }
// For XNNPACK delegate, see also the strong override in // For XNNPACK delegate, see also the strong override in
// lite/enable_xnnpack_delegate.cc. // lite/tflite_with_xnnpack.cc.
TFLITE_ATTRIBUTE_WEAK Interpreter::TfLiteDelegatePtr AcquireXNNPACKDelegate( TFLITE_ATTRIBUTE_WEAK Interpreter::TfLiteDelegatePtr AcquireXNNPACKDelegate(
int num_threads) { int num_threads) {
return Interpreter::TfLiteDelegatePtr(nullptr, [](TfLiteDelegate*) {}); return Interpreter::TfLiteDelegatePtr(nullptr, [](TfLiteDelegate*) {});

View File

@ -18,7 +18,7 @@ limitations under the License.
#include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h" #include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
namespace tflite { namespace tflite {
// Corresponding weak declaration found in lite/model.cc. // Corresponding weak declaration found in lite/interpreter_builder.cc.
std::unique_ptr<TfLiteDelegate, void (*)(TfLiteDelegate*)> std::unique_ptr<TfLiteDelegate, void (*)(TfLiteDelegate*)>
AcquireXNNPACKDelegate(int num_threads) { AcquireXNNPACKDelegate(int num_threads) {
auto opts = TfLiteXNNPackDelegateOptionsDefault(); auto opts = TfLiteXNNPackDelegateOptionsDefault();