TfLiteContext
Summary
Public attributes
|
AcquireSubgraphContext)(struct TfLiteContext *context, int subgraph_index, struct TfLiteContext **acquired_context)
|
TfLiteStatus(*
|
AddTensors)(struct TfLiteContext *, int tensors_to_add, int *first_new_tensor_index)
|
TfLiteStatus(*
|
AllocateBufferForEval)(struct TfLiteContext *ctx, size_t bytes, void **ptr)
|
TfLiteStatus(*
|
AllocatePersistentBuffer)(struct TfLiteContext *ctx, size_t bytes)
|
void *(*
|
GetEvalTensor)(const struct TfLiteContext *context, int tensor_idx)
|
|
GetExecutionPlan)(struct TfLiteContext *context, TfLiteIntArray **execution_plan)
|
TfLiteStatus(*
|
GetExternalContext)(struct TfLiteContext *, TfLiteExternalContextType)
|
|
GetModelMetadata)(const struct TfLiteContext *context, const char *name, const char **ptr, size_t *bytes)
|
TfLiteStatus(*
|
GetNodeAndRegistration)(struct TfLiteContext *, int node_index, TfLiteNode **node, struct TfLiteRegistration **registration)
|
TfLiteStatus(*
|
GetScratchBuffer)(struct TfLiteContext *ctx, int buffer_idx)
|
void *(*
|
GetTensor)(const struct TfLiteContext *context, int tensor_idx)
|
|
PreviewDelegatePartitioning)(struct TfLiteContext *context, const TfLiteIntArray *nodes_to_replace, TfLiteDelegateParams **partition_params_array, int *num_partitions)
|
TfLiteStatus(*
|
ReleaseSubgraphContext)(struct TfLiteContext *context, int subgraph_index)
|
TfLiteStatus(*
|
ReplaceNodeSubsetsWithDelegateKernels)(struct TfLiteContext *, struct TfLiteRegistration registration, const TfLiteIntArray *nodes_to_replace, struct TfLiteDelegate *delegate)
|
TfLiteStatus(*
|
ReportError)(struct TfLiteContext *, const char *msg,...)
|
void(*
|
RequestScratchBufferInArena)(struct TfLiteContext *ctx, size_t bytes, int *buffer_idx)
|
TfLiteStatus(*
|
ResizeTensor)(struct TfLiteContext *, TfLiteTensor *tensor, TfLiteIntArray *new_size)
|
TfLiteStatus(*
|
ResizeTensorExplicit)(struct TfLiteContext *ctx, TfLiteTensor *tensor, int dims, const int *shape)
|
TfLiteStatus(*
|
SetExternalContext)(struct TfLiteContext *, TfLiteExternalContextType, TfLiteExternalContext *)
|
void(*
|
allow_fp32_relax_to_fp16
|
bool
|
impl_
|
void *
|
profiler
|
void *
|
recommended_num_threads
|
int
|
tensors
|
|
tensors_size
|
size_t
|
Public attributes
AcquireSubgraphContext
TfLiteStatus(* TfLiteContext::AcquireSubgraphContext)(struct TfLiteContext *context, int subgraph_index, struct TfLiteContext **acquired_context)
AddTensors
TfLiteStatus(* TfLiteContext::AddTensors)(struct TfLiteContext *, int tensors_to_add, int *first_new_tensor_index)
AllocateBufferForEval
TfLiteStatus(* TfLiteContext::AllocateBufferForEval)(struct TfLiteContext *ctx, size_t bytes, void **ptr)
AllocatePersistentBuffer
void *(* TfLiteContext::AllocatePersistentBuffer)(struct TfLiteContext *ctx, size_t bytes)
GetEvalTensor
TfLiteEvalTensor *(* TfLiteContext::GetEvalTensor)(const struct TfLiteContext *context, int tensor_idx)
GetExecutionPlan
TfLiteStatus(* TfLiteContext::GetExecutionPlan)(struct TfLiteContext *context, TfLiteIntArray **execution_plan)
GetExternalContext
TfLiteExternalContext *(* TfLiteContext::GetExternalContext)(struct TfLiteContext *, TfLiteExternalContextType)
TfLiteStatus(* TfLiteContext::GetModelMetadata)(const struct TfLiteContext *context, const char *name, const char **ptr, size_t *bytes)
GetNodeAndRegistration
TfLiteStatus(* TfLiteContext::GetNodeAndRegistration)(struct TfLiteContext *, int node_index, TfLiteNode **node, struct TfLiteRegistration **registration)
GetScratchBuffer
void *(* TfLiteContext::GetScratchBuffer)(struct TfLiteContext *ctx, int buffer_idx)
GetTensor
TfLiteTensor *(* TfLiteContext::GetTensor)(const struct TfLiteContext *context, int tensor_idx)
PreviewDelegatePartitioning
TfLiteStatus(* TfLiteContext::PreviewDelegatePartitioning)(struct TfLiteContext *context, const TfLiteIntArray *nodes_to_replace, TfLiteDelegateParams **partition_params_array, int *num_partitions)
ReleaseSubgraphContext
TfLiteStatus(* TfLiteContext::ReleaseSubgraphContext)(struct TfLiteContext *context, int subgraph_index)
ReplaceNodeSubsetsWithDelegateKernels
TfLiteStatus(* TfLiteContext::ReplaceNodeSubsetsWithDelegateKernels)(struct TfLiteContext *, struct TfLiteRegistration registration, const TfLiteIntArray *nodes_to_replace, struct TfLiteDelegate *delegate)
ReportError
void(* TfLiteContext::ReportError)(struct TfLiteContext *, const char *msg,...)
RequestScratchBufferInArena
TfLiteStatus(* TfLiteContext::RequestScratchBufferInArena)(struct TfLiteContext *ctx, size_t bytes, int *buffer_idx)
ResizeTensor
TfLiteStatus(* TfLiteContext::ResizeTensor)(struct TfLiteContext *, TfLiteTensor *tensor, TfLiteIntArray *new_size)
ResizeTensorExplicit
TfLiteStatus(* TfLiteContext::ResizeTensorExplicit)(struct TfLiteContext *ctx, TfLiteTensor *tensor, int dims, const int *shape)
SetExternalContext
void(* TfLiteContext::SetExternalContext)(struct TfLiteContext *, TfLiteExternalContextType, TfLiteExternalContext *)
allow_fp32_relax_to_fp16
bool TfLiteContext::allow_fp32_relax_to_fp16
impl_
void * TfLiteContext::impl_
profiler
void * TfLiteContext::profiler
recommended_num_threads
int TfLiteContext::recommended_num_threads
tensors
TfLiteTensor * TfLiteContext::tensors
tensors_size
size_t TfLiteContext::tensors_size