/external/tensorflow/tensorflow/lite/c/ |
D | c_api_internal.h | 60 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/ |
D | elementwise.cc | 38 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 …]
|
D | kernel_util.h | 30 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,
|
D | gemm_support.cc | 30 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()
|
D | reshape.cc | 31 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()
|
D | logical.cc | 37 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()
|
D | eigen_support.cc | 112 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()
|
D | dequantize.cc | 32 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()
|
D | unique.cc | 29 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()
|
D | pooling.cc | 53 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 …]
|
D | reduce.cc | 49 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 …]
|
D | gemm_support.h | 38 gemmlowp::GemmContext* GetFromContext(TfLiteContext* context); 42 void IncrementUsageCounter(TfLiteContext* context); 46 void DecrementUsageCounter(TfLiteContext* context);
|
D | eigen_support.h | 29 void IncrementUsageCounter(TfLiteContext* context); 33 void DecrementUsageCounter(TfLiteContext* context); 41 TfLiteContext* context);
|
D | squared_difference.cc | 43 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()
|
D | unpack.cc | 36 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()
|
D | floor_div.cc | 43 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/ |
D | subgraph.h | 224 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/ |
D | kernel_test.cc | 30 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/ |
D | fully_connected.cc | 48 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()
|
D | depthwise_conv.cc | 50 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/ |
D | node_info_delegate.h | 31 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;
|
D | calibrator.cc | 115 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()
|
D | logging_op_resolver_test.cc | 25 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/ |
D | simple_memory_arena.h | 53 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/ |
D | micro_mutable_op_resolver_test.cc | 22 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()
|