Skip to content

Commit d17fd00

Browse files
ggawryalcopybara-github
authored andcommitted
Pass allocation info to InterpreterBuilder in InterpreterCreateWithOpResolver.
PiperOrigin-RevId: 828415520
1 parent 49ad0a5 commit d17fd00

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

tflite/core/c/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,7 @@ tflite_cc_library_with_c_headers_test(
112112
"//tflite:version",
113113
"//tflite/c:c_api_internal",
114114
"//tflite/c:common_internal",
115+
"//tflite/converter:allocation",
115116
"//tflite/core:create_op_resolver_header",
116117
"//tflite/core:framework_stable",
117118
"//tflite/core:signature_runner",

tflite/core/c/c_api.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -571,7 +571,8 @@ TfLiteInterpreter* InterpreterCreateWithOpResolver(
571571
? optional_error_reporter.get()
572572
: tflite::DefaultErrorReporter();
573573
tflite::InterpreterBuilder builder(model->impl->GetModel(), *op_resolver,
574-
error_reporter);
574+
error_reporter, nullptr,
575+
model->impl->allocation());
575576

576577
if (optional_options && optional_options->telemetry_profiler) {
577578
std::unique_ptr<tflite::telemetry::TelemetryProfiler> profiler;

0 commit comments

Comments
 (0)