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
// lite/enable_xnnpack_delegate.cc.
// lite/tflite_with_xnnpack.cc.
TFLITE_ATTRIBUTE_WEAK Interpreter::TfLiteDelegatePtr AcquireXNNPACKDelegate(
int num_threads) {
return Interpreter::TfLiteDelegatePtr(nullptr, [](TfLiteDelegate*) {});

View File

@ -18,7 +18,7 @@ limitations under the License.
#include "tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h"
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*)>
AcquireXNNPACKDelegate(int num_threads) {
auto opts = TfLiteXNNPackDelegateOptionsDefault();