Home
last modified time | relevance | path

Searched refs:TfLiteContext (Results 1 – 25 of 101) sorted by relevance

12345

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dcommon.h63 struct TfLiteContext;
74 TfLiteStatus (*Refresh)(struct TfLiteContext* context);
636 typedef struct TfLiteContext { struct
703 struct TfLiteContext*, int node_index, TfLiteNode** node, argument
806 } TfLiteContext; typedef
827 void (*free)(TfLiteContext* context, void* buffer);
845 const char* (*profiling_string)(const TfLiteContext* context,
914 TfLiteStatus (*Prepare)(TfLiteContext* context,
921 TfLiteStatus (*CopyFromBufferHandle)(TfLiteContext* context,
928 TfLiteStatus (*CopyToBufferHandle)(TfLiteContext* context,
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/
A Dkernel_util.h40 const TfLiteTensor* GetInput(const TfLiteContext* context,
71 TfLiteTensor* GetOutput(TfLiteContext* context, const TfLiteNode* node,
118 TfLiteStatus GetTemporarySafe(const TfLiteContext* context,
129 const TfLiteTensor* GetIntermediates(TfLiteContext* context,
142 TfLiteStatus GetIntermediatesSafe(const TfLiteContext* context,
213 TfLiteContext* context, const TfLiteTensor* input,
220 TfLiteContext* context, const TfLiteTensor* input,
229 TfLiteStatus GetQuantizedConvolutionMultipler(TfLiteContext* context,
236 TfLiteStatus GetQuantizedConvolutionMultipler(TfLiteContext* context,
275 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context,
[all …]
A Dkernel_util.cc42 inline TfLiteTensor* GetTensorAtIndex(const TfLiteContext* context, in GetTensorAtIndex()
57 TF_LITE_KERNEL_LOG(const_cast<TfLiteContext*>(context), in ValidateTensorIndexingSafe()
63 TF_LITE_KERNEL_LOG(const_cast<TfLiteContext*>(context), in ValidateTensorIndexingSafe()
86 inline TfLiteTensor* GetMutableInput(const TfLiteContext* context, in GetMutableInput()
109 const TfLiteTensor* GetInput(const TfLiteContext* context, in GetInput()
161 TfLiteStatus GetTemporarySafe(const TfLiteContext* context, in GetTemporarySafe()
172 const TfLiteTensor* GetIntermediates(TfLiteContext* context, in GetIntermediates()
182 TfLiteStatus GetIntermediatesSafe(const TfLiteContext* context, in GetIntermediatesSafe()
196 TfLiteContext* context, const TfLiteTensor* input, in PopulateConvolutionQuantizationParams()
211 TfLiteContext* context, const TfLiteTensor* input, in PopulateConvolutionQuantizationParams()
[all …]
A Deigen_support.h29 void IncrementUsageCounter(TfLiteContext* context);
33 void DecrementUsageCounter(TfLiteContext* context);
41 TfLiteContext* context);
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dkernel_runner.h49 static TfLiteTensor* GetTensor(const struct TfLiteContext* context,
51 static TfLiteEvalTensor* GetEvalTensor(const struct TfLiteContext* context,
53 static void* AllocatePersistentBuffer(TfLiteContext* context, size_t bytes);
54 static TfLiteStatus RequestScratchBufferInArena(TfLiteContext* context,
57 static void* GetScratchBuffer(TfLiteContext* context, int buffer_index);
58 static void ReportOpError(struct TfLiteContext* context, const char* format,
71 TfLiteContext context_ = {};
A Delementwise.cc40 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare()
57 inline TfLiteStatus EvalImpl(TfLiteContext* context, TfLiteNode* node, in EvalImpl()
71 inline TfLiteStatus EvalNumeric(TfLiteContext* context, TfLiteNode* node, in EvalNumeric()
81 TfLiteStatus AbsEval(TfLiteContext* context, TfLiteNode* node) { in AbsEval()
85 TfLiteStatus SinEval(TfLiteContext* context, TfLiteNode* node) { in SinEval()
89 TfLiteStatus CosEval(TfLiteContext* context, TfLiteNode* node) { in CosEval()
93 TfLiteStatus LogEval(TfLiteContext* context, TfLiteNode* node) { in LogEval()
97 TfLiteStatus SqrtEval(TfLiteContext* context, TfLiteNode* node) { in SqrtEval()
101 TfLiteStatus RsqrtEval(TfLiteContext* context, TfLiteNode* node) { in RsqrtEval()
105 TfLiteStatus SquareEval(TfLiteContext* context, TfLiteNode* node) { in SquareEval()
[all …]
A Dkernel_util.h31 inline TfLiteEvalTensor* GetMutableEvalInput(const TfLiteContext* context, in GetMutableEvalInput()
40 inline const TfLiteEvalTensor* GetEvalInput(const TfLiteContext* context, in GetEvalInput()
46 inline TfLiteEvalTensor* GetEvalOutput(const TfLiteContext* context, in GetEvalOutput()
79 TfLiteStatus CreateWritableTensorDimsWithCopy(TfLiteContext* context,
A Dkernel_runner.cc76 TfLiteTensor* KernelRunner::GetTensor(const struct TfLiteContext* context, in GetTensor()
86 const struct TfLiteContext* context, int tensor_index) { in GetEvalTensor()
104 void* KernelRunner::AllocatePersistentBuffer(TfLiteContext* context, in AllocatePersistentBuffer()
113 TfLiteStatus KernelRunner::RequestScratchBufferInArena(TfLiteContext* context, in RequestScratchBufferInArena()
140 void* KernelRunner::GetScratchBuffer(TfLiteContext* context, int buffer_index) { in GetScratchBuffer()
152 void KernelRunner::ReportOpError(struct TfLiteContext* context, in ReportOpError()
A Dfloor_mod.cc38 TfLiteStatus CalculateOpData(TfLiteContext* context, TfLiteNode* node) { in CalculateOpData()
58 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
62 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
67 TfLiteStatus EvalFloorMod(TfLiteContext* context, bool requires_broadcast, in EvalFloorMod()
92 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
A Dsvdf.h50 void EvalIntegerSvdfReference(TfLiteContext* context, TfLiteNode* node,
61 TfLiteContext* context, TfLiteNode* node, const TfLiteEvalTensor* input,
67 TfLiteStatus PrepareSvdf(TfLiteContext* context, TfLiteNode* node);
A Dfloor_div.cc33 TfLiteStatus CalculateOpData(TfLiteContext* context, TfLiteNode* node) { in CalculateOpData()
53 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
57 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
62 TfLiteStatus EvalFloorDiv(TfLiteContext* context, in EvalFloorDiv()
97 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
A Dfill.cc30 TfLiteStatus EnsureEqImpl(TfLiteContext* context, const TfLiteIntArray* array, in EnsureEqImpl()
40 TfLiteStatus EnsureEq(TfLiteContext* context, const TfLiteIntArray* array, in EnsureEq()
69 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
99 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
A Dsoftmax.h24 void* SoftmaxInit(TfLiteContext* context, const char* buffer, size_t length);
26 TfLiteStatus SoftmaxPrepare(TfLiteContext* context, TfLiteNode* node);
A Dconcatenation.cc53 inline void GetAllInputTensorShapes(const TfLiteContext* context, in GetAllInputTensorShapes()
75 inline void GetAllInputTensorData(const TfLiteContext* context, in GetAllInputTensorData()
87 void EvalUnquantized(TfLiteContext* context, TfLiteNode* node) { in EvalUnquantized()
107 void EvalQuantizedUInt8(TfLiteContext* context, TfLiteNode* node) { in EvalQuantizedUInt8()
128 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
133 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
229 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
A Dadd_n.cc48 TfLiteStatus CalculateOpData(TfLiteContext* context, TfLiteNode* node) { in CalculateOpData()
129 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
134 inline const T** CopyInputsToScratchBuffer(TfLiteContext* context, in CopyInputsToScratchBuffer()
150 void EvalAddN(TfLiteContext* context, TfLiteNode* node, in EvalAddN()
164 void EvalAddNQuantized(TfLiteContext* context, TfLiteNode* node, in EvalAddNQuantized()
187 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
A Dquantize.h33 TfLiteStatus EvalQuantizeReference(TfLiteContext* context, TfLiteNode* node);
34 TfLiteStatus PrepareQuantizeReference(TfLiteContext* context, TfLiteNode* node);
A Dlogistic.cc43 TfLiteStatus CalculateArithmeticOpData(TfLiteContext* context, TfLiteNode* node, in CalculateArithmeticOpData()
72 void* LogisticInit(TfLiteContext* context, const char* buffer, size_t length) { in LogisticInit()
77 TfLiteStatus LogisticPrepare(TfLiteContext* context, TfLiteNode* node) { in LogisticPrepare()
84 TfLiteStatus LogisticEval(TfLiteContext* context, TfLiteNode* node) { in LogisticEval()
A Dlogical.cc32 TfLiteStatus LogicalImpl(TfLiteContext* context, TfLiteNode* node, in LogicalImpl()
64 TfLiteStatus LogicalOrEval(TfLiteContext* context, TfLiteNode* node) { in LogicalOrEval()
70 TfLiteStatus LogicalAndEval(TfLiteContext* context, TfLiteNode* node) { in LogicalAndEval()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/
A Dtest_helpers.h62 static void* Init(TfLiteContext* context, const char* buffer, size_t length);
63 static TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node);
64 static TfLiteStatus Invoke(TfLiteContext* context, TfLiteNode* node);
72 static void Free(TfLiteContext* context, void* buffer);
73 static TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node);
74 static TfLiteStatus Invoke(TfLiteContext* context, TfLiteNode* node);
86 static void Free(TfLiteContext* context, void* buffer);
87 static TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node);
88 static TfLiteStatus Invoke(TfLiteContext* context, TfLiteNode* node);
150 void ReportOpError(struct TfLiteContext* context, const char* format, ...);
[all …]
A Dmicro_interpreter.h133 const TfLiteContext& context() const { return context_; } in context()
144 static void* AllocatePersistentBuffer(TfLiteContext* ctx, size_t bytes);
145 static TfLiteStatus RequestScratchBufferInArena(TfLiteContext* ctx,
148 static void* GetScratchBuffer(TfLiteContext* ctx, int buffer_idx);
149 static void ReportOpError(struct TfLiteContext* context, const char* format,
151 static TfLiteTensor* GetTensor(const struct TfLiteContext* context,
153 static TfLiteEvalTensor* GetEvalTensor(const struct TfLiteContext* context,
155 static TfLiteStatus GetGraph(struct TfLiteContext* context,
161 TfLiteContext context_ = {};
A Dmicro_interpreter.cc312 void* MicroInterpreter::AllocatePersistentBuffer(TfLiteContext* ctx, in AllocatePersistentBuffer()
318 TfLiteStatus MicroInterpreter::RequestScratchBufferInArena(TfLiteContext* ctx, in RequestScratchBufferInArena()
327 void* MicroInterpreter::GetScratchBuffer(TfLiteContext* ctx, int buffer_idx) { in GetScratchBuffer()
335 void MicroInterpreter::ReportOpError(struct TfLiteContext* context, in ReportOpError()
347 TfLiteTensor* MicroInterpreter::GetTensor(const struct TfLiteContext* context, in GetTensor()
357 const struct TfLiteContext* context, int tensor_idx) { in GetEvalTensor()
365 TfLiteStatus MicroInterpreter::GetGraph(struct TfLiteContext* context, in GetGraph()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/
A Dsimple_memory_arena.h74 TfLiteStatus Allocate(TfLiteContext* context, size_t alignment, size_t size,
78 TfLiteStatus Deallocate(TfLiteContext* context,
88 TfLiteStatus Commit(TfLiteContext* context);
90 TfLiteStatus ResolveAlloc(TfLiteContext* context,
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/resource/
A Dlookup_interfaces.h35 virtual TfLiteStatus Lookup(TfLiteContext* context, const TfLiteTensor* keys,
38 virtual TfLiteStatus Import(TfLiteContext* context, const TfLiteTensor* keys,
44 virtual TfLiteStatus CheckKeyAndValueTypes(TfLiteContext* context,
A Dstatic_hashtable.h43 TfLiteStatus Lookup(TfLiteContext* context, const TfLiteTensor* keys,
48 TfLiteStatus Import(TfLiteContext* context, const TfLiteTensor* keys,
57 TfLiteStatus CheckKeyAndValueTypes(TfLiteContext* context, in CheckKeyAndValueTypes()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Dpooling.cc47 TfLiteStatus CalculateOpData(TfLiteContext* context, in CalculateOpData()
79 void AverageEvalFloat(const TfLiteContext* context, const TfLiteNode* node, in AverageEvalFloat()
101 void AverageEvalQuantized(TfLiteContext* context, const TfLiteNode* node, in AverageEvalQuantized()
173 void MaxEvalFloat(TfLiteContext* context, TfLiteNode* node, in MaxEvalFloat()
194 void MaxEvalQuantizedUInt8(TfLiteContext* context, TfLiteNode* node, in MaxEvalQuantizedUInt8()
213 TfLiteStatus MaxEvalInt8(TfLiteContext* context, const TfLiteNode* node, in MaxEvalInt8()
266 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
271 TfLiteStatus MaxPrepare(TfLiteContext* context, TfLiteNode* node) { in MaxPrepare()
288 TfLiteStatus AveragePrepare(TfLiteContext* context, TfLiteNode* node) { in AveragePrepare()
323 TfLiteStatus AverageEval(TfLiteContext* context, TfLiteNode* node) { in AverageEval()
[all …]

Completed in 28 milliseconds

12345