| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 68 const ::tflite::BuiltinOperator op, const OperatorSignature& op_signature) { in GetVersioningOpSig()
90 : public BuiltinOperator<AveragePoolOperator, ::tflite::Pool2DOptions,
93 using BuiltinOperator::BuiltinOperator;
119 : public BuiltinOperator<ConvOperator, ::tflite::Conv2DOptions,
122 using BuiltinOperator::BuiltinOperator;
149 : public BuiltinOperator<DepthwiseConvOperator,
153 using BuiltinOperator::BuiltinOperator;
192 class Add : public BuiltinOperator<AddOperator, ::tflite::AddOptions,
195 using BuiltinOperator::BuiltinOperator;
212 class AddN : public BuiltinOperator<AddNOperator, ::tflite::AddNOptions,
[all …]
|
| D | builtin_operator.h | 33 class BuiltinOperator : public BaseOperator {
38 BuiltinOperator(::tflite::BuiltinOperator op, OperatorType type) in BuiltinOperator() function
75 ::tflite::BuiltinOperator builtin_op() const { return builtin_op_; } in builtin_op()
78 const ::tflite::BuiltinOperator builtin_op_;
|
| D | simple_operator.h | 36 SimpleOperator(::tflite::BuiltinOperator op, OperatorType type) in SimpleOperator()
55 ::tflite::BuiltinOperator builtin_op() const { return builtin_op_; } in builtin_op()
58 const ::tflite::BuiltinOperator builtin_op_;
|
| D | export.cc | 38 using ::tflite::BuiltinOperator;
86 const std::map<std::string, BuiltinOperator>& GetBuiltinOpsMap() { in GetBuiltinOpsMap()
87 static std::map<std::string, BuiltinOperator>* builtin_ops = nullptr; in GetBuiltinOpsMap()
89 builtin_ops = new std::map<std::string, BuiltinOperator>(); in GetBuiltinOpsMap()
92 BuiltinOperator op = static_cast<BuiltinOperator>(i); in GetBuiltinOpsMap()
299 std::map<std::string, BuiltinOperator> builtin_ops; in ExportOperatorCodes()
301 BuiltinOperator op = static_cast<BuiltinOperator>(i); in ExportOperatorCodes()
|
| D | export.h | 103 OperatorKey(::tflite::BuiltinOperator type, const std::string& custom_code, in OperatorKey()
109 ::tflite::BuiltinOperator type() const { return type_; } in type()
154 ::tflite::BuiltinOperator type_ = ::tflite::BuiltinOperator_CUSTOM;
|
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver.h | 40 struct ValueHasher<tflite::BuiltinOperator> {
41 size_t operator()(const tflite::BuiltinOperator& v) const {
64 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
67 void AddBuiltin(tflite::BuiltinOperator op,
69 void AddBuiltin(tflite::BuiltinOperator op,
79 typedef std::pair<tflite::BuiltinOperator, int> BuiltinOperatorKey;
|
| D | mutable_op_resolver.cc | 27 const TfLiteRegistration* MutableOpResolver::FindOp(tflite::BuiltinOperator op, in FindOp()
39 void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin()
50 void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_utils.cc | 45 BuiltinOperator GetBuiltinCode(const OperatorCode* op_code) { in GetBuiltinCode()
51 static_cast<BuiltinOperator>(op_code->deprecated_builtin_code())); in GetBuiltinCode()
54 BuiltinOperator GetBuiltinCode(const OperatorCodeT* op_code) { in GetBuiltinCode()
58 return std::max(op_code->builtin_code, static_cast<BuiltinOperator>( in GetBuiltinCode()
|
| D | schema_conversion_utils.h | 24 const BuiltinOperator builtin_code);
30 BuiltinOperator builtin_code = BuiltinOperator_ADD,
36 BuiltinOperator builtin_code = BuiltinOperator_ADD,
|
| D | schema_conversion_utils.cc | 24 const BuiltinOperator builtin_code) { in ConvertBuiltinCodeToDeprecatedBuiltinCode()
41 flatbuffers::FlatBufferBuilder &_fbb, BuiltinOperator builtin_code, in CreateOperatorCode()
58 flatbuffers::FlatBufferBuilder &_fbb, BuiltinOperator builtin_code, in CreateOperatorCodeDirect()
|
| D | schema_utils.h | 27 BuiltinOperator GetBuiltinCode(const OperatorCode *op_code);
29 BuiltinOperator GetBuiltinCode(const OperatorCodeT *op_code);
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_op_resolver.h | 45 virtual const TfLiteRegistration* FindOp(BuiltinOperator op) const = 0;
53 const TfLiteRegistration* FindOp(BuiltinOperator op, in FindOp()
66 virtual BuiltinParseFunction GetOpDataParser(BuiltinOperator op) const = 0;
|
| D | micro_mutable_op_resolver.h | 44 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override { in FindOp()
68 BuiltinOperator op) const override { in GetOpDataParser()
449 TfLiteStatus AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin()
499 BuiltinOperator builtin_codes_[tOpCount];
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | min_max_builder_test.cc | 25 MinMaxOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MinMaxOpModel()
35 MinMaxOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MinMaxOpModel()
80 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
98 void TestModelConstInput(tflite::BuiltinOperator op, const TensorData& input1, in TestModelConstInput()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | logging_op_resolver.h | 45 const TfLiteRegistration* FindOp(BuiltinOperator op,
47 KernelEvalFuncPtr GetWrappedKernelInvoke(BuiltinOperator op,
|
| D | calibration_common.h | 26 using BuiltinOperatorKey = std::pair<BuiltinOperator, int>;
59 BuiltinOperator builtin_op_code;
|
| D | logging_op_resolver.cc | 80 const TfLiteRegistration* LoggingOpResolver::FindOp(BuiltinOperator op, in FindOp()
91 KernelEvalFuncPtr LoggingOpResolver::GetWrappedKernelInvoke(BuiltinOperator op, in GetWrappedKernelInvoke()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | unary_elementwise_tester.h | 55 void Test(tflite::BuiltinOperator unary_op, TfLiteDelegate* delegate) const;
58 std::vector<char> CreateTfLiteModel(tflite::BuiltinOperator unary_op) const;
|
| D | binary_elementwise_tester.h | 116 void Test(tflite::BuiltinOperator binary_op, TfLiteDelegate* delegate) const;
119 std::vector<char> CreateTfLiteModel(tflite::BuiltinOperator binary_op) const;
|
| D | reduce_tester.h | 100 void Test(tflite::BuiltinOperator reduce_op, TfLiteDelegate* delegate) const;
103 std::vector<char> CreateTfLiteModel(tflite::BuiltinOperator reduce_op) const;
|
| D | pool_2d_tester.h | 150 void Test(tflite::BuiltinOperator pool_op, TfLiteDelegate* delegate) const;
153 std::vector<char> CreateTfLiteModel(tflite::BuiltinOperator pool_op) const;
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | runtime_version.cc | 50 std::string FindMinimumRuntimeVersionForOp(tflite::BuiltinOperator op_code, in FindMinimumRuntimeVersionForOp()
55 static const std::map<std::pair<BuiltinOperator, int>, std::string>* in FindMinimumRuntimeVersionForOp()
57 new std::map<std::pair<BuiltinOperator, int>, std::string>({ in FindMinimumRuntimeVersionForOp()
351 std::pair<BuiltinOperator, int> version_key = {op_code, op_version}; in FindMinimumRuntimeVersionForOp()
|
| D | runtime_version.h | 29 std::string FindMinimumRuntimeVersionForOp(tflite::BuiltinOperator op_code,
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | logical_test.cc | 32 std::initializer_list<int> input2_shape, BuiltinOperator op) { in LogicalOpModel()
51 void ConfigureBuiltinOp(BuiltinOperator op) { in ConfigureBuiltinOp()
|
| D | maximum_minimum_test.cc | 34 MaxMinOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MaxMinOpModel()
44 MaxMinOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MaxMinOpModel()
75 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
|