Do not use TfLitePtrUnion.uint8, it's deprecated; use .data instead.
PiperOrigin-RevId: 306493152 Change-Id: I1805afc68e4ffb37a076bec79e0e51eac9110336
This commit is contained in:
parent
7d2ff41266
commit
60140d9369
@ -83,10 +83,10 @@ TfLiteStatus AllocateVariables(
|
|||||||
TfLiteTensor* runtime_tensors, SimpleMemoryAllocator* allocator) {
|
TfLiteTensor* runtime_tensors, SimpleMemoryAllocator* allocator) {
|
||||||
for (size_t i = 0; i < flatbuffer_tensors->size(); ++i) {
|
for (size_t i = 0; i < flatbuffer_tensors->size(); ++i) {
|
||||||
if (flatbuffer_tensors->Get(i)->is_variable()) {
|
if (flatbuffer_tensors->Get(i)->is_variable()) {
|
||||||
runtime_tensors[i].data.uint8 = allocator->AllocateFromTail(
|
runtime_tensors[i].data.data = allocator->AllocateFromTail(
|
||||||
runtime_tensors[i].bytes, kBufferAlignment);
|
runtime_tensors[i].bytes, kBufferAlignment);
|
||||||
// Allocation failure.
|
// Allocation failure.
|
||||||
if (runtime_tensors[i].data.uint8 == nullptr) {
|
if (runtime_tensors[i].data.data == nullptr) {
|
||||||
return kTfLiteError;
|
return kTfLiteError;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user