Home
last modified time | relevance | path

Searched refs:TfLiteRegistration (Results 1 – 25 of 91) sorted by relevance

1234

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/
A Dbuiltin_op_kernels.h35 TfLiteRegistration* Register_ABS();
36 TfLiteRegistration* Register_ADD();
37 TfLiteRegistration* Register_ADD_N();
47 TfLiteRegistration* Register_CAST();
48 TfLiteRegistration* Register_CEIL();
53 TfLiteRegistration* Register_COS();
59 TfLiteRegistration* Register_DIV();
60 TfLiteRegistration* Register_ELU();
64 TfLiteRegistration* Register_EXP();
82 TfLiteRegistration* Register_IF();
[all …]
A Dcustom_ops_register.h24 TfLiteRegistration* Register_ATAN2();
25 TfLiteRegistration* Register_HASHTABLE();
26 TfLiteRegistration* Register_HASHTABLE_FIND();
27 TfLiteRegistration* Register_HASHTABLE_IMPORT();
28 TfLiteRegistration* Register_HASHTABLE_SIZE();
29 TfLiteRegistration* Register_IRFFT2D();
30 TfLiteRegistration* Register_MULTINOMIAL();
31 TfLiteRegistration* Register_RANDOM_STANDARD_NORMAL();
32 TfLiteRegistration* Register_RANDOM_UNIFORM();
33 TfLiteRegistration* Register_RANDOM_UNIFORM_INT();
[all …]
A Dfully_connected.h29 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();
A Dvariable_ops.h25 TfLiteRegistration* Register_READ_VARIABLE();
26 TfLiteRegistration* Register_ASSIGN_VARIABLE();
A Dregister_ref.h30 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
32 const TfLiteRegistration* FindOp(const char* op, int version) const override;
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dmicro_ops.h34 TfLiteRegistration Register_ADD_N();
36 TfLiteRegistration Register_CAST();
41 TfLiteRegistration Register_DIV();
42 TfLiteRegistration Register_ELU();
43 TfLiteRegistration Register_EXP();
45 TfLiteRegistration Register_FILL();
65 TfLiteRegistration Register_ABS();
66 TfLiteRegistration Register_ADD();
74 TfLiteRegistration Register_COS();
83 TfLiteRegistration Register_LOG();
[all …]
A Dconv_test.h31 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);
A Delementwise.cc116 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()
A Dfully_connected.h66 TfLiteRegistration Register_FULLY_CONNECTED();
76 TfLiteRegistration Register_FULLY_CONNECTED_INT8();
84 inline TfLiteRegistration Register_FULLY_CONNECTED_INT8() { in Register_FULLY_CONNECTED_INT8()
A Dkernel_runner.h34 KernelRunner(const TfLiteRegistration& registration, TfLiteTensor* tensors,
68 const TfLiteRegistration& registration_;
A Dlogical.cc77 TfLiteRegistration Register_LOGICAL_OR() { in Register_LOGICAL_OR()
90 TfLiteRegistration Register_LOGICAL_AND() { in Register_LOGICAL_AND()
A Darg_min_max.cc109 TfLiteRegistration Register_ARG_MAX() { in Register_ARG_MAX()
120 TfLiteRegistration Register_ARG_MIN() { in Register_ARG_MIN()
A Dethosu.cc23 TfLiteRegistration* Register_ETHOSU() { return nullptr; } in Register_ETHOSU()
A Dethosu.h22 TfLiteRegistration* Register_ETHOSU();
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/
A Dmutable_op_resolver.h64 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
66 const TfLiteRegistration* FindOp(const char* op, int version) const override;
71 const TfLiteRegistration* registration, int version = 1);
77 const TfLiteRegistration* registration, int min_version,
82 void AddCustom(const char* name, const TfLiteRegistration* registration,
88 void AddCustom(const char* name, const TfLiteRegistration* registration,
112 std::unordered_map<BuiltinOperatorKey, TfLiteRegistration,
115 std::unordered_map<CustomOperatorKey, TfLiteRegistration,
A Dutil.h87 TfLiteRegistration CreateUnresolvedCustomOp(const char* custom_op_name);
90 bool IsUnresolvedCustomOp(const TfLiteRegistration& registration);
93 std::string GetOpNameByRegistration(const TfLiteRegistration& registration);
A Dinterpreter_builder.h127 std::vector<const TfLiteRegistration*> flatbuffer_op_index_to_registration_;
128 std::vector<TfLiteRegistration> unresolved_custom_ops_;
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/
A Dmicro_op_resolver.h45 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()
A Dtest_helpers.h60 static const TfLiteRegistration* getRegistration();
61 static TfLiteRegistration* GetMutableRegistration();
69 static const TfLiteRegistration* getRegistration();
70 static TfLiteRegistration* GetMutableRegistration();
83 static const TfLiteRegistration* getRegistration();
84 static TfLiteRegistration* GetMutableRegistration();
A Dmicro_mutable_op_resolver.h34 TfLiteRegistration* Register_DETECTION_POSTPROCESS();
44 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override { in FindOp()
48 const TfLiteRegistration& registration = registrations_[i]; in FindOp()
56 const TfLiteRegistration* FindOp(const char* op) const override { in FindOp()
58 const TfLiteRegistration& registration = registrations_[i]; in FindOp()
82 TfLiteStatus AddCustom(const char* name, TfLiteRegistration* registration) { in AddCustom()
103 TfLiteRegistration* new_registration = &registrations_[registrations_len_]; in AddCustom()
216 TfLiteRegistration* registration = tflite::Register_ETHOSU(); in AddEthosU()
252 const TfLiteRegistration& registration = Register_FULLY_CONNECTED()) {
518 const TfLiteRegistration& registration, in AddBuiltin()
[all …]
A Dmicro_graph.cc30 const char* OpNameFromRegistration(const TfLiteRegistration* registration) { in OpNameFromRegistration()
65 const TfLiteRegistration* registration = in InitSubgraphs()
100 const TfLiteRegistration* registration = in PrepareSubgraphs()
130 const TfLiteRegistration* registration = in FreeSubgraphs()
160 const TfLiteRegistration* registration = subgraph_allocations_[subgraph_idx] in InvokeSubgraph()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dc_api_experimental.h48 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,
A Dc_api_internal.h55 const TfLiteRegistration* (*find_builtin_op)(void* user_data,
61 const TfLiteRegistration* (*find_custom_op)(void* user_data, const char* op,
A Dcommon.h65 struct TfLiteRegistration;
704 struct TfLiteRegistration** registration);
709 struct TfLiteContext*, struct TfLiteRegistration registration,
808 typedef struct TfLiteRegistration { struct
865 } TfLiteRegistration; argument
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/core/api/
A Dop_resolver.h33 virtual const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
36 virtual const TfLiteRegistration* FindOp(const char* op,
73 const TfLiteRegistration** registration);

Completed in 26 milliseconds

1234