| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | op_validator.h | 25 const TfLiteNode* node, TfLiteContext* context);
27 const TfLiteNode* node, TfLiteContext* context);
29 const TfLiteNode* node,
32 const TfLiteNode* node,
35 const TfLiteNode* node, TfLiteContext* context);
37 const TfLiteNode* node, TfLiteContext* context);
39 const TfLiteNode* node, TfLiteContext* context);
41 const TfLiteNode* node, TfLiteContext* context,
44 const TfLiteNode* node,
47 const TfLiteNode* node,
|
| D | op_builder.h | 59 OpBuilder* AddBuilder(int builtin_code, const TfLiteNode* node);
63 const TfLiteNode* node);
131 void SetTfLiteNode(const TfLiteNode* node);
161 const TfLiteNode* tflite_node_ = nullptr;
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.h | 41 const TfLiteNode* node, int index);
51 TfLiteStatus GetInputSafe(const TfLiteContext* context, const TfLiteNode* node,
61 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node,
71 TfLiteTensor* GetOutput(TfLiteContext* context, const TfLiteNode* node,
82 TfLiteStatus GetOutputSafe(const TfLiteContext* context, const TfLiteNode* node,
95 const TfLiteNode* node, int index);
105 TfLiteTensor* GetTemporary(TfLiteContext* context, const TfLiteNode* node,
119 const TfLiteNode* node, int index,
130 const TfLiteNode* node, int index);
143 const TfLiteNode* node, int index,
[all …]
|
| D | reshape.cc | 34 TfLiteIntArray* GetOutputShape(TfLiteContext*, TfLiteNode*);
36 TfLiteStatus ResizeOutput(TfLiteContext* context, TfLiteNode* node) { in ResizeOutput()
74 TfLiteNode* node) { in GetOutputShapeFromTensor()
87 TfLiteNode* node) { in GetOutputShapeFromParam()
108 inline bool ShapeIsVector(TfLiteContext* context, TfLiteNode* node) { in ShapeIsVector()
114 TfLiteIntArray* GetOutputShape(TfLiteContext* context, TfLiteNode* node) { in GetOutputShape()
122 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
144 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
|
| D | elementwise.cc | 84 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare()
136 inline TfLiteStatus EvalImpl(TfLiteContext* context, TfLiteNode* node, in EvalImpl()
158 inline TfLiteStatus EvalImpl(TfLiteContext* context, TfLiteNode* node, in EvalImpl()
165 inline TfLiteStatus EvalNumeric(TfLiteContext* context, TfLiteNode* node, in EvalNumeric()
170 inline TfLiteStatus EvalLogical(TfLiteContext* context, TfLiteNode* node, in EvalLogical()
185 TfLiteStatus AbsEvalQuantized(TfLiteContext* context, TfLiteNode* node, in AbsEvalQuantized()
206 TfLiteStatus AbsEval(TfLiteContext* context, TfLiteNode* node) { in AbsEval()
222 TfLiteStatus SinEval(TfLiteContext* context, TfLiteNode* node) { in SinEval()
226 TfLiteStatus CosEval(TfLiteContext* context, TfLiteNode* node) { in CosEval()
230 TfLiteStatus LogEval(TfLiteContext* context, TfLiteNode* node) { in LogEval()
[all …]
|
| D | kernel_util.cc | 81 const TfLiteNode* node, int index) { in GetMutableInput()
91 const TfLiteNode* node, int index, in GetMutableInputSafe()
104 const TfLiteNode* node, int index) { in GetInput()
108 TfLiteStatus GetInputSafe(const TfLiteContext* context, const TfLiteNode* node, in GetInputSafe()
113 TfLiteTensor* GetVariableInput(TfLiteContext* context, const TfLiteNode* node, in GetVariableInput()
119 TfLiteTensor* GetOutput(TfLiteContext* context, const TfLiteNode* node, in GetOutput()
129 TfLiteStatus GetOutputSafe(const TfLiteContext* context, const TfLiteNode* node, in GetOutputSafe()
140 const TfLiteNode* node, int index) { in GetOptionalInputTensor()
145 TfLiteTensor* GetTemporary(TfLiteContext* context, const TfLiteNode* node, in GetTemporary()
156 const TfLiteNode* node, int index, in GetTemporarySafe()
[all …]
|
| D | split.cc | 32 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext()
42 TfLiteStatus UseDynamicOutputTensors(TfLiteContext* context, TfLiteNode* node) { in UseDynamicOutputTensors()
51 TfLiteStatus ResizeOutputTensors(TfLiteContext* context, TfLiteNode* node, in ResizeOutputTensors()
78 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
106 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
|
| D | pooling.cc | 68 TfLiteStatus GenericPrepare(TfLiteContext* context, TfLiteNode* node) { in GenericPrepare()
116 void AverageEvalFloat(TfLiteContext* context, TfLiteNode* node, in AverageEvalFloat()
144 void AverageEvalQuantizedUint8(TfLiteContext* context, TfLiteNode* node, in AverageEvalQuantizedUint8()
174 void AverageEvalQuantizedInt8(TfLiteContext* context, TfLiteNode* node, in AverageEvalQuantizedInt8()
204 void AverageEvalQuantizedInt16(TfLiteContext* context, TfLiteNode* node, in AverageEvalQuantizedInt16()
230 void MaxEvalFloat(TfLiteContext* context, TfLiteNode* node, in MaxEvalFloat()
257 void MaxEvalQuantizedUInt8(TfLiteContext* context, TfLiteNode* node, in MaxEvalQuantizedUInt8()
286 void MaxEvalQuantizedInt8(TfLiteContext* context, TfLiteNode* node, in MaxEvalQuantizedInt8()
315 void MaxEvalQuantizedInt16(TfLiteContext* context, TfLiteNode* node, in MaxEvalQuantizedInt16()
340 void L2EvalFloat(TfLiteContext* context, TfLiteNode* node, in L2EvalFloat()
[all …]
|
| D | maximum_minimum.cc | 45 OpContext(TfLiteContext* context, TfLiteNode* node) { in OpContext()
55 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { in Prepare()
94 void TFLiteOperation(TfLiteContext* context, TfLiteNode* node, in TFLiteOperation()
109 TfLiteContext* context, TfLiteNode* node, const OpContext& op_context) { in TFLiteOperation()
134 TfLiteContext* context, TfLiteNode* node, const OpContext& op_context) { in TFLiteOperation()
157 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | elementwise.cc | 40 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()
76 inline TfLiteStatus EvalLogical(TfLiteContext* context, TfLiteNode* node, in EvalLogical()
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()
[all …]
|
| D | kernel_util.h | 32 const TfLiteNode* node, in GetMutableEvalInput()
41 const TfLiteNode* node, int index) { in GetEvalInput()
47 const TfLiteNode* node, int index) { in GetEvalOutput()
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | blacklist_base.h | 64 TfLiteNode* node) = 0;
70 TfLiteNode* node) = 0;
82 TfLiteStatus Resize(TfLiteContext* context, TfLiteNode* node);
84 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node);
|
| D | string_projection_base.h | 115 TfLiteNode* node) = 0;
121 TfLiteNode* node) = 0;
146 TfLiteStatus Resize(TfLiteContext* context, TfLiteNode* node);
148 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 76 absl::Status RetrieveBuiltinData(const TfLiteNode* tflite_node, in RetrieveBuiltinData()
175 const TfLiteNode* tflite_node, in IsSupported()
187 absl::Status Parse(const TfLiteNode* tflite_node, in Parse()
209 const TfLiteNode* tflite_node, in IsSupported()
215 absl::Status Parse(const TfLiteNode* tflite_node, in Parse()
230 const TfLiteNode* tflite_node, in IsSupported()
244 absl::Status Parse(const TfLiteNode* tflite_node, in Parse()
347 const TfLiteNode* tflite_node, in IsSupported()
374 absl::Status Parse(const TfLiteNode* tflite_node, in Parse()
407 const TfLiteNode* tflite_node, in IsSupported()
[all …]
|
| D | model_builder_helper.h | 41 TfLiteNode** tflite_node,
61 const TfLiteNode* tflite_node);
64 const TfLiteNode* tflite_node);
67 const TfLiteNode* tflite_node,
71 const TfLiteNode* tflite_node,
|
| D | operation_parser.h | 36 virtual absl::Status Parse(const TfLiteNode* tflite_node,
42 const TfLiteNode* tflite_node,
59 const TfLiteNode* tflite_node, int idx);
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | logging_op_resolver_test.cc | 26 TfLiteStatus ConvPrepare(TfLiteContext* context, TfLiteNode* node) { in ConvPrepare()
30 TfLiteStatus ConvEval(TfLiteContext* context, TfLiteNode* node) { in ConvEval()
34 TfLiteStatus AddPrepare(TfLiteContext* context, TfLiteNode* node) { in AddPrepare()
38 TfLiteStatus AddEval(TfLiteContext* context, TfLiteNode* node) { in AddEval()
42 TfLiteStatus CustomPrepare(TfLiteContext* context, TfLiteNode* node) { in CustomPrepare()
46 TfLiteStatus CustomEval(TfLiteContext* context, TfLiteNode* node) { in CustomEval()
50 TfLiteStatus WrappingInvoke(TfLiteContext* context, TfLiteNode* node) { in WrappingInvoke()
|
| D | calibrator.cc | 60 Calibrator(const std::unordered_map<const TfLiteNode*, OperatorInfo>& in Calibrator() argument
71 KernelEvalFuncPtr GetKernelInvoke(const TfLiteNode* node) const;
80 const OperatorInfo& GetOpInfo(const TfLiteNode* node) const { in GetOpInfo()
85 std::unordered_map<const TfLiteNode*, OperatorInfo> node_ptr_opinfo_map_;
92 KernelEvalFuncPtr Calibrator::GetKernelInvoke(const TfLiteNode* node) const { in GetKernelInvoke()
149 const std::unordered_map<const TfLiteNode*, OperatorInfo>& node_to_opinfo, in CreateCalibrator() argument
178 TfLiteNode* node, in GetLoggingEvalFunc()
197 TfLiteStatus LoggingEval(TfLiteContext* context, TfLiteNode* node) { in LoggingEval()
279 std::unordered_map<const TfLiteNode*, OperatorInfo>* node_ptr_opinfo_map, in GetNodeOpInfoMapAndContext() argument
407 std::unordered_map<const TfLiteNode*, OperatorInfo> node_ptr_opinfo_map; in BuildLoggingInterpreter()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test_util.cc | 49 reg.prepare = [](TfLiteContext* context, TfLiteNode* node) { in AddOpRegistration()
69 reg.invoke = [](TfLiteContext* context, TfLiteNode* node) { in AddOpRegistration()
150 TfLiteNode* node; in SimpleDelegate()
165 TfLiteNode* node; in SimpleDelegate()
254 reg.invoke = [](TfLiteContext* context, TfLiteNode* node) -> TfLiteStatus { in FakeFusedRegistration()
258 reg.invoke = [](TfLiteContext* context, TfLiteNode* node) -> TfLiteStatus { in FakeFusedRegistration()
289 reg.prepare = [](TfLiteContext* context, TfLiteNode* node) { in FakeFusedRegistration()
304 reg.prepare = [](TfLiteContext* context, TfLiteNode* node) { in FakeFusedRegistration()
308 reg.prepare = [](TfLiteContext* context, TfLiteNode* node) { in FakeFusedRegistration()
430 [=](TfLiteContext* context, TfLiteNode* node, in FP16Delegate()
[all …]
|
| D | utils.h | 45 std::function<bool(TfLiteContext*, TfLiteNode*, TfLiteRegistration*,
95 virtual bool IsNodeSupported(TfLiteContext* context, TfLiteNode* node, in IsNodeSupported()
148 bool IsNodeSupported(TfLiteContext* context, TfLiteNode* node,
162 void RemapFp16InputTensors(TfLiteNode* node,
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | test_helpers.h | 63 static TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node);
64 static TfLiteStatus Invoke(TfLiteContext* context, TfLiteNode* node);
73 static TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node);
74 static TfLiteStatus Invoke(TfLiteContext* context, TfLiteNode* node);
87 static TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node);
88 static TfLiteStatus Invoke(TfLiteContext* context, TfLiteNode* node);
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 449 typedef struct TfLiteNode { struct
481 } TfLiteNode; argument
543 typedef struct TfLiteNode { struct
561 } TfLiteNode; argument
665 struct TfLiteContext*, int node_index, TfLiteNode** node,
796 TfLiteStatus (*prepare)(TfLiteContext* context, TfLiteNode* node);
800 TfLiteStatus (*invoke)(TfLiteContext* context, TfLiteNode* node);
808 const TfLiteNode* node);
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.h | 52 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) override;
55 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) override;
76 TfLiteStatus ResizeOutputTensors(TfLiteContext* context, TfLiteNode* node);
|
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate.h | 55 virtual TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) = 0;
59 virtual TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) = 0;
87 const TfLiteNode* node,
|
| /external/tensorflow/tensorflow/lite/delegates/utils/dummy_delegate/ |
| D | dummy_delegate.cc | 35 TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) override { in Prepare()
39 TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) override { in Eval()
54 const TfLiteNode* node, in IsNodeSupportedByDelegate()
|