Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/lite/c/
Dc_api_internal.h60 TfLiteStatus (*Refresh)(struct TfLiteContext* context);
393 typedef struct TfLiteContext { struct
412 TfLiteStatus (*GetExecutionPlan)(struct TfLiteContext* context, argument
423 TfLiteStatus (*ResizeTensor)(struct TfLiteContext*, TfLiteTensor* tensor, argument
426 void (*ReportError)(struct TfLiteContext*, const char* msg, ...); argument
431 TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add, argument
436 TfLiteStatus (*GetNodeAndRegistration)(struct TfLiteContext*, int node_index, argument
443 struct TfLiteContext*, TfLiteRegistration registration, argument
452 TfLiteExternalContext* (*GetExternalContext)(struct TfLiteContext*, argument
457 void (*SetExternalContext)(struct TfLiteContext*, TfLiteExternalContextType, argument
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Delementwise.cc38 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare()
54 inline TfLiteStatus EvalImpl(TfLiteContext* context, TfLiteNode* node, in EvalImpl()
68 inline TfLiteStatus EvalNumeric(TfLiteContext* context, TfLiteNode* node, in EvalNumeric()
73 inline TfLiteStatus EvalLogical(TfLiteContext* context, TfLiteNode* node, in EvalLogical()
78 TfLiteStatus AbsEval(TfLiteContext* context, TfLiteNode* node) { in AbsEval()
82 TfLiteStatus SinEval(TfLiteContext* context, TfLiteNode* node) { in SinEval()
86 TfLiteStatus CosEval(TfLiteContext* context, TfLiteNode* node) { in CosEval()
90 TfLiteStatus LogEval(TfLiteContext* context, TfLiteNode* node) { in LogEval()
94 TfLiteStatus SqrtEval(TfLiteContext* context, TfLiteNode* node) { in SqrtEval()
98 TfLiteStatus RsqrtEval(TfLiteContext* context, TfLiteNode* node) { in RsqrtEval()
[all …]
Dkernel_util.h30 inline const TfLiteTensor* GetInput(TfLiteContext* context, TfLiteNode* node, in GetInput()
34 inline TfLiteTensor* GetVariableInput(TfLiteContext* context, TfLiteNode* node, in GetVariableInput()
39 inline TfLiteTensor* GetOutput(TfLiteContext* context, TfLiteNode* node, in GetOutput()
43 inline TfLiteTensor* GetTemporary(TfLiteContext* context, TfLiteNode* node, in GetTemporary()
58 inline const TfLiteTensor* GetOptionalInputTensor(TfLiteContext* context, in GetOptionalInputTensor()
89 TfLiteContext* context, const TfLiteTensor* input,
102 TfLiteStatus GetQuantizedConvolutionMultipler(TfLiteContext* context,
111 TfLiteStatus CalculateActivationRangeQuantized(TfLiteContext* context,
147 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context,
Dgemm_support.cc30 RefCountedGemmContext* GetGemmLowpContext(TfLiteContext* context) { in GetGemmLowpContext()
35 TfLiteStatus Refresh(TfLiteContext* context) { in Refresh()
45 void IncrementUsageCounter(TfLiteContext* context) { in IncrementUsageCounter()
61 void DecrementUsageCounter(TfLiteContext* context) { in DecrementUsageCounter()
74 gemmlowp::GemmContext* GetFromContext(TfLiteContext* context) { in GetFromContext()
Dreshape.cc31 TfLiteStatus ResizeOutput(TfLiteContext* context, TfLiteNode* node, in ResizeOutput()
62 TfLiteIntArray* GetOutputShapeFromTensor(TfLiteContext* context, in GetOutputShapeFromTensor()
74 TfLiteIntArray* GetOutputShapeFromParam(TfLiteContext* context, in GetOutputShapeFromParam()
96 bool ShapeIsVector(TfLiteContext* context, TfLiteNode* node) { in ShapeIsVector()
101 TfLiteIntArray* GetOutputShape(TfLiteContext* context, TfLiteNode* node) { in GetOutputShape()
109 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
130 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dlogical.cc37 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
43 void Free(TfLiteContext* context, void* buffer) { in Free()
47 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
80 TfLiteStatus LogicalImpl(TfLiteContext* context, TfLiteNode* node, in LogicalImpl()
103 TfLiteStatus LogicalOrEval(TfLiteContext* context, TfLiteNode* node) { in LogicalOrEval()
108 TfLiteStatus LogicalAndEval(TfLiteContext* context, TfLiteNode* node) { in LogicalAndEval()
Deigen_support.cc112 RefCountedEigenContext* GetEigenContext(TfLiteContext* context) { in GetEigenContext()
117 TfLiteStatus Refresh(TfLiteContext* context) { in Refresh()
130 void IncrementUsageCounter(TfLiteContext* context) { in IncrementUsageCounter()
147 void DecrementUsageCounter(TfLiteContext* context) { in DecrementUsageCounter()
160 const Eigen::ThreadPoolDevice* GetThreadPoolDevice(TfLiteContext* context) { in GetThreadPoolDevice()
Ddequantize.cc32 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext()
45 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
51 void Free(TfLiteContext* context, void* buffer) { in Free()
55 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
74 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dunique.cc29 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
33 void Free(TfLiteContext* context, void* buffer) {} in Free()
35 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
61 TfLiteStatus EvalImpl(TfLiteContext* context, const TfLiteTensor* input, in EvalImpl()
98 TfLiteStatus EvalImpl(TfLiteContext* context, const TfLiteTensor* input, in EvalImpl()
121 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dpooling.cc53 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
60 void Free(TfLiteContext* context, void* buffer) { in Free()
65 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare()
123 void AverageEvalFloat(TfLiteContext* context, TfLiteNode* node, in AverageEvalFloat()
151 void AverageEvalQuantizedUint8(TfLiteContext* context, TfLiteNode* node, in AverageEvalQuantizedUint8()
180 void AverageEvalQuantizedInt8(TfLiteContext* context, TfLiteNode* node, in AverageEvalQuantizedInt8()
202 void MaxEvalFloat(TfLiteContext* context, TfLiteNode* node, in MaxEvalFloat()
229 void MaxEvalQuantizedUInt8(TfLiteContext* context, TfLiteNode* node, in MaxEvalQuantizedUInt8()
258 void MaxEvalQuantizedInt8(TfLiteContext* context, TfLiteNode* node, in MaxEvalQuantizedInt8()
283 void L2EvalFloat(TfLiteContext* context, TfLiteNode* node, in L2EvalFloat()
[all …]
Dreduce.cc49 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext()
61 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
70 void Free(TfLiteContext* context, void* buffer) { in Free()
76 TfLiteStatus ResizeTempAxis(TfLiteContext* context, OpContext* op_context, in ResizeTempAxis()
84 TfLiteStatus ResizeTempSum(TfLiteContext* context, OpContext* op_context, in ResizeTempSum()
92 TfLiteStatus ResizeOutputTensor(TfLiteContext* context, OpContext* op_context) { in ResizeOutputTensor()
160 TfLiteStatus InitializeTemporaries(TfLiteContext* context, TfLiteNode* node, in InitializeTemporaries()
207 TfLiteStatus PrepareSimple(TfLiteContext* context, TfLiteNode* node) { in PrepareSimple()
228 TfLiteStatus PrepareAny(TfLiteContext* context, TfLiteNode* node) { in PrepareAny()
235 TfLiteStatus PrepareMeanOrSum(TfLiteContext* context, TfLiteNode* node) { in PrepareMeanOrSum()
[all …]
Dgemm_support.h38 gemmlowp::GemmContext* GetFromContext(TfLiteContext* context);
42 void IncrementUsageCounter(TfLiteContext* context);
46 void DecrementUsageCounter(TfLiteContext* context);
Deigen_support.h29 void IncrementUsageCounter(TfLiteContext* context);
33 void DecrementUsageCounter(TfLiteContext* context);
41 TfLiteContext* context);
Dsquared_difference.cc43 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
49 void Free(TfLiteContext* context, void* buffer) { in Free()
53 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
80 void EvalSquaredDifference(TfLiteContext* context, TfLiteNode* node, in EvalSquaredDifference()
96 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dunpack.cc36 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
42 void Free(TfLiteContext* context, void* buffer) { in Free()
46 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
91 void UnpackImpl(TfLiteContext* context, TfLiteNode* node, in UnpackImpl()
102 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Dfloor_div.cc43 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
49 void Free(TfLiteContext* context, void* buffer) { in Free()
53 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
87 TfLiteStatus EvalImpl(TfLiteContext* context, bool requires_broadcast, in EvalImpl()
114 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h224 TfLiteContext* context() { return context_; } in context()
370 static TfLiteStatus ResizeTensor(TfLiteContext* context, TfLiteTensor* tensor,
373 static void ReportErrorC(TfLiteContext* context, const char* format, ...);
376 static TfLiteStatus AddTensors(TfLiteContext* context, int tensors_to_add,
382 TfLiteContext* context, TfLiteRegistration registration,
401 static TfLiteStatus GetNodeAndRegistration(struct TfLiteContext*,
413 static TfLiteStatus GetExecutionPlan(struct TfLiteContext* context,
419 struct TfLiteContext* context, TfLiteExternalContextType type);
422 static void SetExternalContext(struct TfLiteContext* context,
466 TfLiteContext owned_context_;
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel_test.cc30 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteDelegate* delegate, in GenericPrepare()
61 typedef TfLiteStatus (*PrepareFunction)(TfLiteContext* context,
67 return [](TfLiteContext* context, TfLiteDelegate* delegate) { in GetPrepareFunction()
78 delegate_.CopyFromBufferHandle = [](TfLiteContext* context, in ConfigureDelegate()
112 ConfigureDelegate([](TfLiteContext* context, TfLiteDelegate* delegate) { in TEST_F()
143 ConfigureDelegate([](TfLiteContext* context, TfLiteDelegate* delegate) { in TEST_F()
156 ConfigureDelegate([](TfLiteContext* context, TfLiteDelegate* delegate) { in TEST_F()
174 ConfigureDelegate([](TfLiteContext* context, TfLiteDelegate* delegate) { in TEST_F()
192 ConfigureDelegate([](TfLiteContext* context, TfLiteDelegate* delegate) { in TEST_F()
210 ConfigureDelegate([](TfLiteContext* context, TfLiteDelegate* delegate) { in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dfully_connected.cc48 TfLiteStatus CalculateOpData(TfLiteContext* context, in CalculateOpData()
71 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
75 void Free(TfLiteContext* context, void* buffer) {} in Free()
77 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
81 TfLiteStatus EvalQuantized(TfLiteContext* context, TfLiteNode* node, in EvalQuantized()
124 TfLiteStatus EvalFloat(TfLiteContext* context, TfLiteNode* node, in EvalFloat()
142 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
Ddepthwise_conv.cc50 TfLiteStatus CalculateOpData(TfLiteContext* context, TfLiteNode* node, in CalculateOpData()
84 void* Init(TfLiteContext* context, const char* buffer, size_t length) { in Init()
88 void Free(TfLiteContext* context, void* buffer) {} in Free()
90 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
94 void EvalFloat(TfLiteContext* context, TfLiteNode* node, in EvalFloat()
122 void EvalQuantized(TfLiteContext* context, TfLiteNode* node, in EvalQuantized()
156 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dnode_info_delegate.h31 virtual TfLiteStatus OnDelegatePrepareCalled(TfLiteContext* context) = 0;
53 TfLiteStatus OnDelegatePrepareCalled(TfLiteContext* context) override;
56 const TfLiteContext* GetContext() const { return context_; } in GetContext()
59 const TfLiteContext* context_ = nullptr;
Dcalibrator.cc115 Calibrator* GetCalibrator(const TfLiteContext* context) const { in GetCalibrator()
124 void RemoveCalibrator(const TfLiteContext* context) { in RemoveCalibrator()
132 const TfLiteContext* context, in CreateCalibrator()
149 std::unordered_map<const TfLiteContext*, std::unique_ptr<Calibrator>>
160 TfLiteStatus LoggingEval(TfLiteContext* context, TfLiteNode* node) { in LoggingEval()
221 const TfLiteContext** context in GetNodeOpInfoMapAndContext()
248 Reader(const TfLiteContext* context, const Logger* logger) in Reader()
254 const TfLiteContext* context_;
329 const TfLiteContext* context = nullptr; in BuildLoggingInterpreter()
Dlogging_op_resolver_test.cc25 TfLiteStatus ConvPrepare(TfLiteContext* context, TfLiteNode* node) { in ConvPrepare()
29 TfLiteStatus ConvEval(TfLiteContext* context, TfLiteNode* node) { in ConvEval()
33 TfLiteStatus AddPrepare(TfLiteContext* context, TfLiteNode* node) { in AddPrepare()
37 TfLiteStatus AddEval(TfLiteContext* context, TfLiteNode* node) { in AddEval()
41 TfLiteStatus WrappingInvoke(TfLiteContext* context, TfLiteNode* node) { in WrappingInvoke()
/external/tensorflow/tensorflow/lite/
Dsimple_memory_arena.h53 TfLiteStatus Allocate(TfLiteContext* context, size_t alignment, size_t size,
56 TfLiteStatus Deallocate(TfLiteContext* context, const ArenaAlloc& alloc);
65 TfLiteStatus Commit(TfLiteContext* context);
67 TfLiteStatus ResolveAlloc(TfLiteContext* context, const ArenaAlloc& alloc,
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_mutable_op_resolver_test.cc22 void* MockInit(TfLiteContext* context, const char* buffer, size_t length) { in MockInit()
27 void MockFree(TfLiteContext* context, void* buffer) { in MockFree()
31 TfLiteStatus MockPrepare(TfLiteContext* context, TfLiteNode* node) { in MockPrepare()
35 TfLiteStatus MockInvoke(TfLiteContext* context, TfLiteNode* node) { in MockInvoke()

1234567