| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 73 const ::tflite::BuiltinOperator op, const OperatorSignature& op_signature) { in GetVersioningOpSig()
101 : public BuiltinOperator<AveragePoolOperator, ::tflite::Pool2DOptions,
104 using BuiltinOperator::BuiltinOperator;
130 : public BuiltinOperator<ConvOperator, ::tflite::Conv2DOptions,
133 using BuiltinOperator::BuiltinOperator;
160 : public BuiltinOperator<DepthwiseConvOperator,
164 using BuiltinOperator::BuiltinOperator;
204 class Add : public BuiltinOperator<AddOperator, ::tflite::AddOptions,
207 using BuiltinOperator::BuiltinOperator;
224 class AddN : public BuiltinOperator<AddNOperator, ::tflite::AddNOptions,
[all …]
|
| D | builtin_operator.h | 35 class BuiltinOperator : public BaseOperator {
40 BuiltinOperator(::tflite::BuiltinOperator op, OperatorType type) in BuiltinOperator() function
77 ::tflite::BuiltinOperator builtin_op() const { return builtin_op_; } in builtin_op()
80 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.h | 107 OperatorKey(::tflite::BuiltinOperator type, const std::string& custom_code, in OperatorKey()
113 ::tflite::BuiltinOperator type() const { return type_; } in type()
158 ::tflite::BuiltinOperator type_ = ::tflite::BuiltinOperator_CUSTOM;
|
| D | export.cc | 40 using ::tflite::BuiltinOperator;
88 const std::map<std::string, BuiltinOperator>& GetBuiltinOpsMap() { in GetBuiltinOpsMap()
89 static std::map<std::string, BuiltinOperator>* builtin_ops = nullptr; in GetBuiltinOpsMap()
91 builtin_ops = new std::map<std::string, BuiltinOperator>(); in GetBuiltinOpsMap()
94 BuiltinOperator op = static_cast<BuiltinOperator>(i); in GetBuiltinOpsMap()
301 std::map<std::string, BuiltinOperator> builtin_ops; in ExportOperatorCodes()
303 BuiltinOperator op = static_cast<BuiltinOperator>(i); in ExportOperatorCodes()
|
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver.h | 41 struct ValueHasher<tflite::BuiltinOperator> {
42 size_t operator()(const tflite::BuiltinOperator& v) const {
65 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
71 void AddBuiltin(tflite::BuiltinOperator op,
77 void AddBuiltin(tflite::BuiltinOperator op,
137 typedef std::pair<tflite::BuiltinOperator, int> BuiltinOperatorKey;
|
| D | mutable_op_resolver.cc | 28 const TfLiteRegistration* MutableOpResolver::FindOp(tflite::BuiltinOperator op, in FindOp()
58 void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin()
81 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/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | unsorted_segment_test.h | 36 const TensorData& num_segments, const BuiltinOperator op, in UnsortedSegmentModel()
52 const BuiltinOperator op, const BuiltinOptions options) { in UnsortedSegmentModel()
77 class UnsortedSegmentTest : public ::testing::TestWithParam<BuiltinOperator> {
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unsorted_segment_test.h | 36 const TensorData& num_segments, const BuiltinOperator op, in UnsortedSegmentModel()
52 const BuiltinOperator op, const BuiltinOptions options) { in UnsortedSegmentModel()
77 class UnsortedSegmentTest : public ::testing::TestWithParam<BuiltinOperator> {
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | min_max_builder_test.cc | 27 MinMaxOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MinMaxOpModel()
37 MinMaxOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MinMaxOpModel()
82 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
100 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 | 27 using BuiltinOperatorKey = std::pair<BuiltinOperator, int>;
61 BuiltinOperator builtin_op_code;
|
| /external/armnn/delegate/test/ |
| D | FillTest.cpp | 19 tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, in Fill2dTest()
37 tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, in Fill3dTest()
64 tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, in Fill4dTest()
99 tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, in FillInt32Test()
|
| D | SoftmaxTestHelper.hpp | 24 std::vector<char> CreateSoftmaxTfLiteModel(tflite::BuiltinOperator softmaxOperatorCode, in CreateSoftmaxTfLiteModel()
105 void SoftmaxTest(tflite::BuiltinOperator softmaxOperatorCode, in SoftmaxTest()
148 void SoftmaxTestCase(tflite::BuiltinOperator operatorCode, in SoftmaxTestCase()
|
| D | ElementwiseUnaryTestHelper.hpp | 24 std::vector<char> CreateElementwiseUnaryTfLiteModel(tflite::BuiltinOperator unaryOperatorCode, in CreateElementwiseUnaryTfLiteModel()
78 void ElementwiseUnaryFP32Test(tflite::BuiltinOperator unaryOperatorCode, in ElementwiseUnaryFP32Test()
112 void ElementwiseUnaryBoolTest(tflite::BuiltinOperator unaryOperatorCode, in ElementwiseUnaryBoolTest()
|
| /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 | quantized_pool_2d_tester.h | 173 void Test(tflite::BuiltinOperator pool_op, Interpreter* delegate_interpreter,
176 void Test(tflite::BuiltinOperator pool_op, TfLiteDelegate* delegate) const;
179 std::vector<char> CreateTfLiteModel(tflite::BuiltinOperator pool_op) const;
|
| D | quantized_unary_elementwise_tester.h | 93 void Test(tflite::BuiltinOperator unary_op, TfLiteDelegate* delegate) const;
96 std::vector<char> CreateTfLiteModel(tflite::BuiltinOperator unary_op) const;
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights_portable.cc | 83 const BuiltinOperator builtin_op_code = GetBuiltinCode(op_code); in GetWeightInputIndices()
138 const BuiltinOperator builtin_op_code = GetBuiltinCode(op_code); in IsHybridEvaluationOp()
298 const BuiltinOperator& op_code = in UpdateInt8OperatorVersions()
331 const BuiltinOperator op_code = in IsQuantizationPassThroughOps()
376 inline bool IsOpDenylisted(const flat_hash_set<BuiltinOperator>& op_denylist, in IsOpDenylisted()
377 const BuiltinOperator op_code) { in IsOpDenylisted()
385 const flat_hash_set<BuiltinOperator>& op_denylist = {}) { in QuantizeWeightsInt8()
660 const flat_hash_set<BuiltinOperator>& op_denylist, in QuantizeWeights()
|
| D | quantize_weights.cc | 97 const BuiltinOperator builtin_op_code = GetBuiltinCode(op_code); in GetWeightInputIndices()
152 const BuiltinOperator builtin_op_code = GetBuiltinCode(op_code); in IsHybridEvaluationOp()
312 const BuiltinOperator& op_code = in UpdateInt8OperatorVersions()
345 const BuiltinOperator op_code = in IsQuantizationPassThroughOps()
390 inline bool IsOpDenylisted(const flat_hash_set<BuiltinOperator>& op_denylist, in IsOpDenylisted()
391 const BuiltinOperator op_code) { in IsOpDenylisted()
399 const flat_hash_set<BuiltinOperator>& op_denylist = {}) { in QuantizeWeightsInt8()
677 const flat_hash_set<BuiltinOperator>& op_denylist, in QuantizeWeights()
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | runtime_version.cc | 47 std::string FindMinimumRuntimeVersionForOp(tflite::BuiltinOperator op_code, in FindMinimumRuntimeVersionForOp()
54 static const std::map<std::pair<BuiltinOperator, int>, in FindMinimumRuntimeVersionForOp()
56 new std::map<std::pair<BuiltinOperator, int>, std::string>( in FindMinimumRuntimeVersionForOp()
391 std::pair<BuiltinOperator, int> version_key = {op_code, op_version}; in FindMinimumRuntimeVersionForOp()
|