Home
last modified time | relevance | path

Searched refs:BuiltinOperator (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator.cc30 : public BuiltinOperator<AveragePoolOperator, ::tflite::Pool2DOptions,
33 using BuiltinOperator::BuiltinOperator;
59 : public BuiltinOperator<ConvOperator, ::tflite::Conv2DOptions,
62 using BuiltinOperator::BuiltinOperator;
85 : public BuiltinOperator<DepthwiseConvOperator,
89 using BuiltinOperator::BuiltinOperator;
113 class Add : public BuiltinOperator<AddOperator, ::tflite::AddOptions,
116 using BuiltinOperator::BuiltinOperator;
134 : public BuiltinOperator<SpaceToBatchNDOperator,
138 using BuiltinOperator::BuiltinOperator;
[all …]
Dbuiltin_operator.h33 class BuiltinOperator : public BaseOperator {
38 BuiltinOperator(::tflite::BuiltinOperator op, OperatorType type) in BuiltinOperator() function
Dexport.cc33 using ::tflite::BuiltinOperator;
174 std::map<string, BuiltinOperator> builtin_ops; in ExportOperatorCodes()
176 BuiltinOperator op = static_cast<BuiltinOperator>(i); in ExportOperatorCodes()
/external/tensorflow/tensorflow/contrib/lite/tools/
Dmutable_op_resolver.h25 struct hash<tflite::BuiltinOperator> {
26 size_t operator()(const tflite::BuiltinOperator& op) const {
43 TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override;
45 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration);
Dmutable_op_resolver.cc21 tflite::BuiltinOperator op) const { in FindOp()
31 void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin()
Dverifier_test.cc44 TfLiteFlatbufferModelBuilder(const std::vector<BuiltinOperator>& builtin_ops, in TfLiteFlatbufferModelBuilder()
70 tflite::BuiltinOperator builtin_op, const char* custom_op) { in AddOperator()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dregister.h29 TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override;
31 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration);
36 size_t operator()(const tflite::BuiltinOperator& x) const { in operator()
40 std::unordered_map<tflite::BuiltinOperator, TfLiteRegistration*,
Dtest_util.h90 SingleOpResolver(const BuiltinOperator op, TfLiteRegistration* registration) in SingleOpResolver()
92 TfLiteRegistration* FindOp(BuiltinOperator op) const override { in FindOp()
101 const BuiltinOperator op_;
142 void SetBuiltinOp(BuiltinOperator type, BuiltinOptions builtin_options_type,
Dregister.cc120 tflite::BuiltinOperator op) const { in FindOp()
130 void BuiltinOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin()
Dpooling_test.cc31 BasePoolingOpModel(BuiltinOperator type, const TensorData& input, in BasePoolingOpModel()
Dactivations_test.cc31 BaseActivationsOpModel(BuiltinOperator type, TensorData input) { in BaseActivationsOpModel()
Dtest_util.cc131 void SingleOpModel::SetBuiltinOp(BuiltinOperator type, in SetBuiltinOp()
/external/tensorflow/tensorflow/contrib/lite/
Dmodel.h121 virtual TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const = 0;
170 std::vector<BuiltinOperator> flatbuffer_op_index_to_registration_types_;
Dnnapi_delegate.cc143 tflite::BuiltinOperator builtin = in AddOpsAndParams()
144 static_cast<tflite::BuiltinOperator>(registration.builtin_code); in AddOpsAndParams()
Dmodel_test.cc58 TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override { in FindOp()
Dmodel.cc205 void* ParseOpData(const Operator* op, BuiltinOperator op_type, in ParseOpData()
/external/tensorflow/tensorflow/contrib/lite/schema/
Dschema_v0.fbs55 enum BuiltinOperator : byte {
204 // An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
207 builtin_code:BuiltinOperator;
Dschema_v2.fbs61 enum BuiltinOperator : byte {
245 // An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
248 builtin_code:BuiltinOperator;
Dschema_v1.fbs60 enum BuiltinOperator : byte {
237 // An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
240 builtin_code:BuiltinOperator;
Dschema_v3.fbs71 enum BuiltinOperator : byte {
255 // An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
258 builtin_code:BuiltinOperator;
Dschema.fbs71 enum BuiltinOperator : byte {
367 // An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
370 builtin_code:BuiltinOperator;
Dschema_generated.h195 enum BuiltinOperator { enum
247 inline BuiltinOperator (&EnumValuesBuiltinOperator())[47] { in EnumValuesBuiltinOperator()
248 static BuiltinOperator values[] = { in EnumValuesBuiltinOperator()
357 inline const char *EnumNameBuiltinOperator(BuiltinOperator e) { in EnumNameBuiltinOperator()
3573 BuiltinOperator builtin_code;
3586 BuiltinOperator builtin_code() const {
3587 return static_cast<BuiltinOperator>(GetField<int8_t>(VT_BUILTIN_CODE, 0));
3607 void add_builtin_code(BuiltinOperator builtin_code) {
3627 BuiltinOperator builtin_code = BuiltinOperator_ADD,
3637 BuiltinOperator builtin_code = BuiltinOperator_ADD,
/external/tensorflow/tensorflow/contrib/lite/g3doc/
Dapis.md223 virtual TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const = 0;
225 virtual void AddOp(tflite::BuiltinOperator op, TfLiteRegistration* registration) = 0;