| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| gemm_support.h | 25 // share a single object, as long as they share a TfLiteContext. The caller
28 // void* Init(TfLiteContext* context, const char*, size_t) {
32 // void Free(TfLiteContext* context, void*) {
35 // TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
38 gemmlowp::GemmContext* GetFromContext(TfLiteContext* context);
42 void IncrementUsageCounter(TfLiteContext* context);
46 void DecrementUsageCounter(TfLiteContext* context);
49 void SetMaxNumThreads(TfLiteContext* context, int num_threads);
|
| gemm_support.cc | 27 void IncrementUsageCounter(TfLiteContext* context) {
38 void DecrementUsageCounter(TfLiteContext* context) {
52 gemmlowp::GemmContext* GetFromContext(TfLiteContext* context) {
61 void SetMaxNumThreads(TfLiteContext* context, int num_threads) {
|
| kernel_util.h | 27 inline TfLiteTensor* GetInput(TfLiteContext* context, TfLiteNode* node,
31 inline TfLiteTensor* GetOutput(TfLiteContext* context, TfLiteNode* node,
46 inline TfLiteTensor* GetOptionalInputTensor(TfLiteContext* context,
78 TfLiteContext* context, TfLiteTensor* input, TfLiteTensor* filter,
95 TfLiteStatus CalculateShapeForBroadcast(TfLiteContext* context,
|
| exp.cc | 35 ExpContext(TfLiteContext* context, TfLiteNode* node) {
43 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
54 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| mean.cc | 35 MeanContext(TfLiteContext* context, TfLiteNode* node) {
47 void* Init(TfLiteContext* context, const char* buffer, size_t length) {
55 void Free(TfLiteContext* context, void* buffer) {
60 TfLiteStatus ResizeTempAxis(TfLiteContext* context, MeanContext* op_context,
68 TfLiteStatus ResizeOutputTensor(TfLiteContext* context,
133 TfLiteStatus InitializeTemporaries(TfLiteContext* context, TfLiteNode* node,
155 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
176 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| split.cc | 31 OpContext(TfLiteContext* context, TfLiteNode* node) {
41 TfLiteStatus UseDynamicOutputTensors(TfLiteContext* context, TfLiteNode* node) {
48 TfLiteStatus ResizeOutputTensors(TfLiteContext* context, TfLiteNode* node,
71 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
95 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| div.cc | 40 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
62 void EvalDivFloat(TfLiteContext* context, TfLiteNode* node,
82 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| squeeze.cc | 29 SqueezeContext(TfLiteContext* context, TfLiteNode* node) {
39 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
81 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| sub.cc | 40 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
62 void EvalSubFloat(TfLiteContext* context, TfLiteNode* node,
82 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| batch_to_space_nd.cc | 37 BatchToSpaceNDContext(TfLiteContext* context, TfLiteNode* node) {
56 TfLiteStatus ResizeOutputTensor(TfLiteContext* context,
95 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
113 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| pad.cc | 37 PadContext(TfLiteContext* context, TfLiteNode* node) {
52 TfLiteStatus ResizeOutputTensor(TfLiteContext* context,
78 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
98 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| space_to_batch_nd.cc | 37 SpaceToBatchNDContext(TfLiteContext* context, TfLiteNode* node) {
56 TfLiteStatus ResizeOutputTensor(TfLiteContext* context,
90 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
108 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| transpose.cc | 35 TransposeContext(TfLiteContext* context, TfLiteNode* node) {
45 TfLiteStatus ResizeOutputTensor(TfLiteContext* context,
68 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
87 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| activations.cc | 44 void* Init(TfLiteContext* context, const char* buffer, size_t length) {
51 void Free(TfLiteContext* context, void* buffer) {
55 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) {
66 TfLiteStatus SigmoidPrepare(TfLiteContext* context, TfLiteNode* node) {
96 TfLiteStatus SoftmaxPrepare(TfLiteContext* context, TfLiteNode* node) {
126 TfLiteStatus ReluEval(TfLiteContext* context, TfLiteNode* node) {
144 TfLiteStatus Relu1Eval(TfLiteContext* context, TfLiteNode* node) {
164 TfLiteStatus Relu6Eval(TfLiteContext* context, TfLiteNode* node) {
182 TfLiteStatus TanhEval(TfLiteContext* context, TfLiteNode* node) {
201 TfLiteStatus SigmoidEval(TfLiteContext* context, TfLiteNode* node)
[all...] |
| pooling.cc | 53 void* Init(TfLiteContext* context, const char* buffer, size_t length) {
60 void Free(TfLiteContext* context, void* buffer) {
65 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) {
123 void AverageEvalFloat(TfLiteContext* context, TfLiteNode* node,
144 void AverageEvalQuantized(TfLiteContext* context, TfLiteNode* node,
167 void MaxEvalFloat(TfLiteContext* context, TfLiteNode* node,
188 void MaxEvalQuantized(TfLiteContext* context, TfLiteNode* node,
211 void L2EvalFloat(TfLiteContext* context, TfLiteNode* node,
234 TfLiteStatus AverageEval(TfLiteContext* context, TfLiteNode* node) {
256 TfLiteStatus MaxEval(TfLiteContext* context, TfLiteNode* node)
[all...] |
| embedding_lookup.cc | 50 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
72 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| l2norm.cc | 37 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
61 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| reshape.cc | 30 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
72 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| add.cc | 44 void* Init(TfLiteContext* context, const char* buffer, size_t length) {
50 void Free(TfLiteContext* context, void* buffer) {
54 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
81 void EvalAddFloat(TfLiteContext* context, TfLiteNode* node,
110 void EvalAddQuantized(TfLiteContext* context, TfLiteNode* node,
163 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| context.h | 23 // TfLiteContext - allows an op to access the tensors
239 typedef struct TfLiteContext {
258 TfLiteStatus (*GetExecutionPlan)(struct TfLiteContext* context,
269 TfLiteStatus (*ResizeTensor)(struct TfLiteContext*, TfLiteTensor* tensor,
272 void (*ReportError)(struct TfLiteContext*, const char* msg, ...);
277 TfLiteStatus (*AddTensors)(struct TfLiteContext*, int tensors_to_add,
282 TfLiteStatus (*GetNodeAndRegistration)(struct TfLiteContext*, int node_index,
288 struct TfLiteContext*, TfLiteRegistration registration,
294 } TfLiteContext;
312 void* (*init)(TfLiteContext* context, const char* buffer, size_t length)
[all...] |
| simple_memory_arena.h | 52 TfLiteStatus Allocate(TfLiteContext* context, size_t alignment, size_t size,
55 TfLiteStatus Deallocate(TfLiteContext* context, const ArenaAlloc& alloc);
64 TfLiteStatus Commit(TfLiteContext* context);
66 TfLiteStatus ResolveAlloc(TfLiteContext* context, const ArenaAlloc& alloc,
|
| arena_planner.h | 47 ArenaPlanner(TfLiteContext* context, std::unique_ptr<GraphInfo> graph_info);
86 TfLiteContext* context_;
|
| simple_memory_arena.cc | 34 TfLiteStatus SimpleMemoryArena::Allocate(TfLiteContext* context,
76 TfLiteStatus SimpleMemoryArena::Deallocate(TfLiteContext* context,
93 TfLiteStatus SimpleMemoryArena::Commit(TfLiteContext* context) {
120 TfLiteStatus SimpleMemoryArena::ResolveAlloc(TfLiteContext* context,
|
| interpreter.h | 330 static TfLiteStatus ResizeTensor(TfLiteContext* context, TfLiteTensor* tensor,
333 static void ReportError(TfLiteContext* context, const char* format, ...);
336 static TfLiteStatus AddTensors(TfLiteContext* context, int tensors_to_add,
342 TfLiteContext* context, TfLiteRegistration registration,
359 static TfLiteStatus GetNodeAndRegistration(struct TfLiteContext*,
370 static TfLiteStatus GetExecutionPlan(struct TfLiteContext* context,
376 TfLiteContext context_;
|
| simple_memory_arena_test.cc | 25 TfLiteContext context;
47 TfLiteContext context;
|
