Fix TFLite makefile build
PiperOrigin-RevId: 260569885
This commit is contained in:
parent
df2fbb8958
commit
01fea718fe
@ -101,7 +101,7 @@ std::vector<std::string> BenchmarkPerformanceOptions::GetValidPerfOptions()
|
|||||||
return {"all", "cpu", "gpu", "nnapi"};
|
return {"all", "cpu", "gpu", "nnapi"};
|
||||||
}
|
}
|
||||||
|
|
||||||
bool BenchmarkPerformanceOptions::HasOption(const string& option) const {
|
bool BenchmarkPerformanceOptions::HasOption(const std::string& option) const {
|
||||||
return std::find(perf_options_.begin(), perf_options_.end(), option) !=
|
return std::find(perf_options_.begin(), perf_options_.end(), option) !=
|
||||||
perf_options_.end();
|
perf_options_.end();
|
||||||
}
|
}
|
||||||
|
@ -49,7 +49,7 @@ class BenchmarkPerformanceOptions {
|
|||||||
|
|
||||||
bool ParsePerfOptions();
|
bool ParsePerfOptions();
|
||||||
virtual std::vector<std::string> GetValidPerfOptions() const;
|
virtual std::vector<std::string> GetValidPerfOptions() const;
|
||||||
bool HasOption(const string& option) const;
|
bool HasOption(const std::string& option) const;
|
||||||
virtual void ResetPerformanceOptions();
|
virtual void ResetPerformanceOptions();
|
||||||
|
|
||||||
virtual void BenchmarkCPUOptions();
|
virtual void BenchmarkCPUOptions();
|
||||||
|
@ -99,6 +99,7 @@ $(wildcard tensorflow/lite/*.c) \
|
|||||||
$(wildcard tensorflow/lite/c/*.c) \
|
$(wildcard tensorflow/lite/c/*.c) \
|
||||||
$(wildcard tensorflow/lite/core/*.cc) \
|
$(wildcard tensorflow/lite/core/*.cc) \
|
||||||
$(wildcard tensorflow/lite/core/api/*.cc) \
|
$(wildcard tensorflow/lite/core/api/*.cc) \
|
||||||
|
tensorflow/lite/experimental/resource_variable/*.cc \
|
||||||
tensorflow/lite/experimental/ruy/allocator.cc \
|
tensorflow/lite/experimental/ruy/allocator.cc \
|
||||||
tensorflow/lite/experimental/ruy/block_map.cc \
|
tensorflow/lite/experimental/ruy/block_map.cc \
|
||||||
tensorflow/lite/experimental/ruy/blocking_counter.cc \
|
tensorflow/lite/experimental/ruy/blocking_counter.cc \
|
||||||
|
Loading…
Reference in New Issue
Block a user