experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TFL & TFLM: Add memory related methods in Context.

Browse Source 
PiperOrigin-RevId: 279037987
Change-Id: Icf65223a4209ace5c2575a91f77ac3a0cfb0966a
This commit is contained in:
Tiezhen WANG 2019-11-07 01:56:52 -08:00 committed by TensorFlower Gardener
parent f45272036c
commit c37435f741
2 changed files with 0 additions and 80 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
tensorflow/lite/c/c_api_internal.h
View File

@ -466,46 +466,6 @@ typedef struct TfLiteContext {
TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add, TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add,
int* first_new_tensor_index); int* first_new_tensor_index);
// Allocate memory for op data. This method should only be used in `Init`
// method and the allocated memory will be available until `Free` method is
// called.
// On TFL, it allocates memory from heap using malloc, but for micro, this
// will be allocating from the allocator.
// WARNING: This is an experimental interface that is subject to change.
void* (*AllocateOpData)(struct TfLiteContext* ctx, size_t size);
// Deallocate memory holding op data. This method should only be used inside
// `Free` method. Caller needs to make sure that that `buffer` is allocated by
// `AllocateOpData` method.
// On TFL, it will free the buffer, and for micro, this method is a no-op.
// WARNING: This is an experimental interface that is subject to change.
void (*DeallocateOpData)(struct TfLiteContext* ctx, void* buffer);
// Allocate a temporary tensor to the node. This method also makes a copy of
// the shape array internally so the shape array could be deallocated right
// afterwards. WARNING: This is an experimental interface that is subject to
// change.
TfLiteStatus (*AllocateTemporaryTensor)(struct TfLiteContext* ctx,
TfLiteNode* node, int dims,
int* shape, TfLiteType data_type,
TfLiteAllocationType allocation_type,
int* new_tensor_index);
// Deallocate all temporary tensors associated to the node (including
// kTfLiteArenaRwPersistent persistent tensors). It also deallocates
// all the shape tensors.
// WARNING: This is an experimental interface that is subject to change.
void (*DeallocateAllTemporaryTensors)(struct TfLiteContext* ctx,
TfLiteNode* node);
// Resize the memory pointer of the `tensor`. This method behaves the same as
// `ResizeTensor`, except that it makes a copy of the shape array internally
// so the shape array could be deallocated right afterwards.
// WARNING: This is an experimental interface that is subject to change.
TfLiteStatus (*ResizeTensorExplicit)(struct TfLiteContext* ctx,
TfLiteTensor* tensor, int dims,
const int* shape);
// Get a Tensor node by node_index. // Get a Tensor node by node_index.
// WARNING: This is an experimental interface that is subject to change. // WARNING: This is an experimental interface that is subject to change.
TfLiteStatus (*GetNodeAndRegistration)( TfLiteStatus (*GetNodeAndRegistration)(

40
tensorflow/lite/experimental/c/c_api_types.h
View File

@ -466,46 +466,6 @@ typedef struct TfLiteContext {
TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add, TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add,
int* first_new_tensor_index); int* first_new_tensor_index);
// Allocate memory for op data. This method should only be used in `Init`
// method and the allocated memory will be available until `Free` method is
// called.
// On TFL, it allocates memory from heap using malloc, but for micro, this
// will be allocating from the allocator.
// WARNING: This is an experimental interface that is subject to change.
void* (*AllocateOpData)(struct TfLiteContext* ctx, size_t size);
// Deallocate memory holding op data. This method should only be used inside
// `Free` method. Caller needs to make sure that that `buffer` is allocated by
// `AllocateOpData` method.
// On TFL, it will free the buffer, and for micro, this method is a no-op.
// WARNING: This is an experimental interface that is subject to change.
void (*DeallocateOpData)(struct TfLiteContext* ctx, void* buffer);
// Allocate a temporary tensor to the node. This method also makes a copy of
// the shape array internally so the shape array could be deallocated right
// afterwards. WARNING: This is an experimental interface that is subject to
// change.
TfLiteStatus (*AllocateTemporaryTensor)(struct TfLiteContext* ctx,
TfLiteNode* node, int dims,
int* shape, TfLiteType data_type,
TfLiteAllocationType allocation_type,
int* new_tensor_index);
// Deallocate all temporary tensors associated to the node (including
// kTfLiteArenaRwPersistent persistent tensors). It also deallocates
// all the shape tensors.
// WARNING: This is an experimental interface that is subject to change.
void (*DeallocateAllTemporaryTensors)(struct TfLiteContext* ctx,
TfLiteNode* node);
// Resize the memory pointer of the `tensor`. This method behaves the same as
// `ResizeTensor`, except that it makes a copy of the shape array internally
// so the shape array could be deallocated right afterwards.
// WARNING: This is an experimental interface that is subject to change.
TfLiteStatus (*ResizeTensorExplicit)(struct TfLiteContext* ctx,
TfLiteTensor* tensor, int dims,
const int* shape);
// Get a Tensor node by node_index. // Get a Tensor node by node_index.
// WARNING: This is an experimental interface that is subject to change. // WARNING: This is an experimental interface that is subject to change.
TfLiteStatus (*GetNodeAndRegistration)( TfLiteStatus (*GetNodeAndRegistration)(