| /external/tensorflow/tensorflow/lite/kernels/ |
| D | builtin_op_kernels.h | 32 TfLiteRegistration* Register_ABS();
33 TfLiteRegistration* Register_ADD();
34 TfLiteRegistration* Register_ADD_N();
35 TfLiteRegistration* Register_ARG_MAX();
36 TfLiteRegistration* Register_ARG_MIN();
37 TfLiteRegistration* Register_AVERAGE_POOL_2D();
38 TfLiteRegistration* Register_BATCH_TO_SPACE_ND();
39 TfLiteRegistration* Register_BATCH_MATMUL();
40 TfLiteRegistration* Register_BIDIRECTIONAL_SEQUENCE_LSTM();
41 TfLiteRegistration* Register_BIDIRECTIONAL_SEQUENCE_RNN();
[all …]
|
| D | register_ref.cc | 28 TfLiteRegistration* Register_NUMERIC_VERIFY_REF();
29 TfLiteRegistration* Register_AUDIO_SPECTROGRAM();
30 TfLiteRegistration* Register_MFCC();
31 TfLiteRegistration* Register_DETECTION_POSTPROCESS();
39 TfLiteRegistration* Register_ABS();
40 TfLiteRegistration* Register_RELU();
41 TfLiteRegistration* Register_RELU_N1_TO_1();
42 TfLiteRegistration* Register_RELU6();
43 TfLiteRegistration* Register_TANH_REF();
44 TfLiteRegistration* Register_LOGISTIC_REF();
[all …]
|
| D | custom_ops_register.h | 24 TfLiteRegistration* Register_HASHTABLE();
25 TfLiteRegistration* Register_HASHTABLE_FIND();
26 TfLiteRegistration* Register_HASHTABLE_IMPORT();
27 TfLiteRegistration* Register_HASHTABLE_SIZE();
28 TfLiteRegistration* Register_MULTINOMIAL();
29 TfLiteRegistration* Register_RANDOM_STANDARD_NORMAL();
30 TfLiteRegistration* Register_RANDOM_UNIFORM();
31 TfLiteRegistration* Register_RANDOM_UNIFORM_INT();
|
| D | fully_connected.h | 29 TfLiteRegistration* Register_FULLY_CONNECTED_REF();
30 TfLiteRegistration* Register_FULLY_CONNECTED_GENERIC_OPT();
31 TfLiteRegistration* Register_FULLY_CONNECTED_PIE();
32 TfLiteRegistration* Register_FULLY_CONNECTED_SPARSE_REF();
33 TfLiteRegistration* Register_FULLY_CONNECTED_SPARSE_OPT();
|
| D | elementwise.cc | 298 TfLiteRegistration* Register_ABS() { in Register_ABS()
299 static TfLiteRegistration r = { in Register_ABS()
308 TfLiteRegistration* Register_SIN() { in Register_SIN()
309 static TfLiteRegistration r = { in Register_SIN()
317 TfLiteRegistration* Register_COS() { in Register_COS()
318 static TfLiteRegistration r = { in Register_COS()
326 TfLiteRegistration* Register_LOG() { in Register_LOG()
327 static TfLiteRegistration r = { in Register_LOG()
335 TfLiteRegistration* Register_SQRT() { in Register_SQRT()
336 static TfLiteRegistration r = { in Register_SQRT()
[all …]
|
| D | maximum_minimum.cc | 194 TfLiteRegistration* Register_MAXIMUM_REF() { in Register_MAXIMUM_REF()
195 static TfLiteRegistration r = { in Register_MAXIMUM_REF()
202 TfLiteRegistration* Register_MAXIMUM_GENERIC_OPT() { in Register_MAXIMUM_GENERIC_OPT()
203 static TfLiteRegistration r = { in Register_MAXIMUM_GENERIC_OPT()
210 TfLiteRegistration* Register_MINIMUM_REF() { in Register_MINIMUM_REF()
211 static TfLiteRegistration r = { in Register_MINIMUM_REF()
218 TfLiteRegistration* Register_MINIMUM_GENERIC_OPT() { in Register_MINIMUM_GENERIC_OPT()
219 static TfLiteRegistration r = { in Register_MINIMUM_GENERIC_OPT()
226 TfLiteRegistration* Register_MAXIMUM() { in Register_MAXIMUM()
229 TfLiteRegistration* Register_MINIMUM() { in Register_MINIMUM()
|
| D | pad.cc | 271 TfLiteRegistration* Register_PAD_REF() { in Register_PAD_REF()
272 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PAD_REF()
277 TfLiteRegistration* Register_PAD_GENERIC_OPT() { in Register_PAD_GENERIC_OPT()
278 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PAD_GENERIC_OPT()
283 TfLiteRegistration* Register_PAD() { return Register_PAD_GENERIC_OPT(); } in Register_PAD()
286 TfLiteRegistration* Register_PADV2_REF() { in Register_PADV2_REF()
287 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PADV2_REF()
292 TfLiteRegistration* Register_PADV2_GENERIC_OPT() { in Register_PADV2_GENERIC_OPT()
293 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PADV2_GENERIC_OPT()
298 TfLiteRegistration* Register_PADV2() { return Register_PADV2_GENERIC_OPT(); } in Register_PADV2()
|
| D | resize_bilinear.cc | 155 TfLiteRegistration* Register_RESIZE_BILINEAR_REF() { in Register_RESIZE_BILINEAR_REF()
156 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_REF()
162 TfLiteRegistration* Register_RESIZE_BILINEAR_GENERIC_OPT() { in Register_RESIZE_BILINEAR_GENERIC_OPT()
163 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_GENERIC_OPT()
169 TfLiteRegistration* Register_RESIZE_BILINEAR_NEON_OPT() { in Register_RESIZE_BILINEAR_NEON_OPT()
170 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_NEON_OPT()
176 TfLiteRegistration* Register_RESIZE_BILINEAR() { in Register_RESIZE_BILINEAR()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | micro_ops.h | 34 TfLiteRegistration Register_BATCH_TO_SPACE_ND();
35 TfLiteRegistration Register_CAST();
36 TfLiteRegistration Register_CONV_2D();
37 TfLiteRegistration Register_DEPTHWISE_CONV_2D();
38 TfLiteRegistration Register_EXP();
39 TfLiteRegistration Register_QUANTIZE();
40 TfLiteRegistration Register_SHAPE();
41 TfLiteRegistration Register_SOFTMAX();
42 TfLiteRegistration Register_SPACE_TO_BATCH_ND();
43 TfLiteRegistration Register_SVDF();
[all …]
|
| D | conv_test.h | 31 TfLiteRegistration registration, float* output_data);
35 TfLiteRegistration registration, int8_t* output_data);
39 TfLiteRegistration registration, uint8_t* output_data);
45 TfLiteRegistration registration,
52 TfLiteRegistration registration,
59 TfLiteRegistration registration,
68 TfLiteRegistration registration, float* output_data);
77 TfLiteConvParams* conv_params, TfLiteRegistration registration,
89 TfLiteRegistration registration, int8_t* output_data);
|
| D | conv_test_common.cc | 24 TfLiteRegistration registration, T* output_data) { in InvokeConv()
46 TfLiteRegistration registration, in ValidateConvGoldens()
62 TfLiteRegistration registration, float* output_data) { in InvokeConv()
69 TfLiteRegistration registration, int8_t* output_data) { in InvokeConv()
76 TfLiteRegistration registration, uint8_t* output_data) { in InvokeConv()
85 TfLiteRegistration registration, in ValidateConvGoldens()
96 TfLiteRegistration registration, in ValidateConvGoldens()
107 TfLiteRegistration registration, in ValidateConvGoldens()
120 TfLiteRegistration registration, in TestConvFloat()
149 TfLiteConvParams* conv_params, TfLiteRegistration registration, in TestConvQuantizedPerLayer()
[all …]
|
| D | elementwise.cc | 116 TfLiteRegistration Register_ABS() { in Register_ABS()
128 TfLiteRegistration Register_SIN() { in Register_SIN()
140 TfLiteRegistration Register_COS() { in Register_COS()
152 TfLiteRegistration Register_LOG() { in Register_LOG()
164 TfLiteRegistration Register_SQRT() { in Register_SQRT()
176 TfLiteRegistration Register_RSQRT() { in Register_RSQRT()
188 TfLiteRegistration Register_SQUARE() { in Register_SQUARE()
200 TfLiteRegistration Register_LOGICAL_NOT() { in Register_LOGICAL_NOT()
|
| /external/libtextclassifier/native/utils/ |
| D | tflite-model-executor.cc | 27 TfLiteRegistration* Register_ADD();
28 TfLiteRegistration* Register_CONCATENATION();
29 TfLiteRegistration* Register_CONV_2D();
30 TfLiteRegistration* Register_EQUAL();
31 TfLiteRegistration* Register_FULLY_CONNECTED();
32 TfLiteRegistration* Register_GREATER_EQUAL();
33 TfLiteRegistration* Register_L2_NORMALIZATION();
34 TfLiteRegistration* Register_MUL();
35 TfLiteRegistration* Register_RESHAPE();
36 TfLiteRegistration* Register_REDUCE_MAX();
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | op_validator.h | 24 bool IsConcatenationOpSupported(const TfLiteRegistration* registration,
26 bool IsConvolutionOpSupported(const TfLiteRegistration* registration,
28 bool IsDepthwiseConvolutionOpSupported(const TfLiteRegistration* registration,
31 bool IsFullyConnectedOpSupported(const TfLiteRegistration* registration,
34 bool IsMeanOpSupported(const TfLiteRegistration* registration,
36 bool IsMirrorPadOpSupported(const TfLiteRegistration* registration,
38 bool IsPadOpSupported(const TfLiteRegistration* registration,
40 bool IsReshapeOpSupported(const TfLiteRegistration* registration,
43 bool IsResizeBilinearOpSupported(const TfLiteRegistration* registration,
46 bool IsTransposeConvolutionOpSupported(const TfLiteRegistration* registration,
|
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver_test.cc | 33 TfLiteRegistration* GetDummyRegistration() { in GetDummyRegistration()
34 static TfLiteRegistration registration = { in GetDummyRegistration()
57 TfLiteRegistration* GetDummy2Registration() { in GetDummy2Registration()
58 static TfLiteRegistration registration = { in GetDummy2Registration()
71 const TfLiteRegistration* found_registration = in TEST()
83 const TfLiteRegistration* found_registration = in TEST()
93 const TfLiteRegistration* found_registration; in TEST()
112 const TfLiteRegistration* found_registration; in TEST()
130 const TfLiteRegistration* found_registration; in TEST()
143 const TfLiteRegistration* found_registration = resolver.FindOp("AWESOME", 1); in TEST()
[all …]
|
| D | mutable_op_resolver.h | 64 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
66 const TfLiteRegistration* FindOp(const char* op, int version) const override;
68 const TfLiteRegistration* registration, int version = 1);
70 const TfLiteRegistration* registration, int min_version,
72 void AddCustom(const char* name, const TfLiteRegistration* registration,
74 void AddCustom(const char* name, const TfLiteRegistration* registration,
82 std::unordered_map<BuiltinOperatorKey, TfLiteRegistration,
85 std::unordered_map<CustomOperatorKey, TfLiteRegistration,
|
| D | mutable_op_resolver.cc | 27 const TfLiteRegistration* MutableOpResolver::FindOp(tflite::BuiltinOperator op, in FindOp()
33 const TfLiteRegistration* MutableOpResolver::FindOp(const char* op, in FindOp()
40 const TfLiteRegistration* registration, in AddBuiltin()
42 TfLiteRegistration new_registration = *registration; in AddBuiltin()
51 const TfLiteRegistration* registration, in AddBuiltin()
59 const TfLiteRegistration* registration, in AddCustom()
61 TfLiteRegistration new_registration = *registration; in AddCustom()
70 const TfLiteRegistration* registration, in AddCustom()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 158 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration = in node()
162 TfLiteRegistration* registration(int index) { in registration()
163 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration = in registration()
165 return const_cast<TfLiteRegistration*>(&node_and_registration->second); in registration()
199 const TfLiteRegistration reg_dequant0 = { in InterpreterFp16()
208 const TfLiteRegistration reg_dequant1 = { in InterpreterFp16()
217 const TfLiteRegistration reg_op0 = { in InterpreterFp16()
309 TfLiteRegistration** registration) { in TEST()
341 TfLiteRegistration* registration = nullptr; in TEST()
368 TfLiteRegistration** registration) { in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops.h | 25 TfLiteRegistration* Register_HASHTABLE();
26 TfLiteRegistration* Register_HASHTABLE_FIND();
27 TfLiteRegistration* Register_HASHTABLE_IMPORT();
28 TfLiteRegistration* Register_HASHTABLE_SIZE();
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | op_resolver_test.cc | 44 const TfLiteRegistration* FindOp(BuiltinOperator op, in FindOp()
47 static TfLiteRegistration r = {MockInit, MockFree, MockPrepare, in FindOp()
54 const TfLiteRegistration* FindOp(const char* op, int version) const override { in FindOp()
56 static TfLiteRegistration r = {MockInit, MockFree, MockPrepare, in FindOp()
87 const TfLiteRegistration* registration = in TEST()
120 const TfLiteRegistration* registration = nullptr; in TEST()
143 const TfLiteRegistration* registration = nullptr; in TEST()
163 const TfLiteRegistration* registration = nullptr; in TEST()
186 const TfLiteRegistration* registration = nullptr; in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.h | 31 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op, in FindOp()
33 static TfLiteRegistration null_registration = {nullptr, nullptr, nullptr, in FindOp()
37 const TfLiteRegistration* FindOp(const char* op, int version) const override { in FindOp()
38 static TfLiteRegistration null_registration = {nullptr, nullptr, nullptr, in FindOp()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | logging_op_resolver.cc | 35 const TfLiteRegistration* base_registration = in LoggingOpResolver()
45 absl::make_unique<TfLiteRegistration>(*base_registration); in LoggingOpResolver()
50 const TfLiteRegistration* base_registration = base_resolver.FindOp( in LoggingOpResolver()
60 absl::make_unique<TfLiteRegistration>(*base_registration); in LoggingOpResolver()
80 const TfLiteRegistration* LoggingOpResolver::FindOp(BuiltinOperator op, in FindOp()
96 const TfLiteRegistration* LoggingOpResolver::FindOp(const char* op, in FindOp()
|
| D | logging_op_resolver.h | 45 const TfLiteRegistration* FindOp(BuiltinOperator op,
50 const TfLiteRegistration* FindOp(const char* op, int version) const override;
54 BuiltinOpsMap<std::unique_ptr<TfLiteRegistration>>
57 CustomOpsMap<std::unique_ptr<TfLiteRegistration>> custom_op_registration_map_;
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_op_resolver.h | 45 virtual const TfLiteRegistration* FindOp(BuiltinOperator op) const = 0;
49 virtual const TfLiteRegistration* FindOp(const char* op) const = 0;
53 const TfLiteRegistration* FindOp(BuiltinOperator op, in FindOp()
60 const TfLiteRegistration* FindOp(const char* op, int version) const final { in FindOp()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_experimental.h | 48 const TfLiteRegistration* registration, int32_t min_version,
67 const TfLiteRegistration* registration, int32_t min_version,
85 const TfLiteRegistration* (*find_builtin_op)(void* user_data,
88 const TfLiteRegistration* (*find_custom_op)(void* user_data,
|