| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 39 : public BuiltinOperator<AveragePoolOperator, ::tflite::Pool2DOptions,
42 using BuiltinOperator::BuiltinOperator;
77 : public BuiltinOperator<ConvOperator, ::tflite::Conv2DOptions,
80 using BuiltinOperator::BuiltinOperator;
130 : public BuiltinOperator<DepthwiseConvOperator,
134 using BuiltinOperator::BuiltinOperator;
183 class Add : public BuiltinOperator<AddOperator, ::tflite::AddOptions,
186 using BuiltinOperator::BuiltinOperator;
213 class AddN : public BuiltinOperator<AddNOperator, ::tflite::AddNOptions,
216 using BuiltinOperator::BuiltinOperator;
[all …]
|
| D | builtin_operator.h | 33 class BuiltinOperator : public BaseOperator {
38 BuiltinOperator(::tflite::BuiltinOperator op, OperatorType type) in BuiltinOperator() function
|
| D | export.cc | 35 using ::tflite::BuiltinOperator;
81 const std::map<string, BuiltinOperator>& GetBuiltinOpsMap() { in GetBuiltinOpsMap()
82 static std::map<string, BuiltinOperator>* builtin_ops = nullptr; in GetBuiltinOpsMap()
84 builtin_ops = new std::map<string, BuiltinOperator>(); in GetBuiltinOpsMap()
87 BuiltinOperator op = static_cast<BuiltinOperator>(i); in GetBuiltinOpsMap()
293 std::map<string, BuiltinOperator> builtin_ops; in ExportOperatorCodes()
295 BuiltinOperator op = static_cast<BuiltinOperator>(i); in ExportOperatorCodes()
|
| D | export.h | 98 OperatorKey(::tflite::BuiltinOperator type, const std::string& custom_code, in OperatorKey()
104 ::tflite::BuiltinOperator type() const { return type_; } in type()
149 ::tflite::BuiltinOperator type_ = ::tflite::BuiltinOperator_CUSTOM;
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | subgraph_quantizer.h | 42 TfLiteStatus QuantizeOpWithBias(BuiltinOperator op_code, OperatorT* op);
46 TfLiteStatus PropagateMinMaxForAvgAndMaxPool(BuiltinOperator op_code,
52 TfLiteStatus AsymmetricQuantizeSoftmax(BuiltinOperator op_code,
56 TfLiteStatus AsymmetricQuantizeInputsAndOutputs(BuiltinOperator op_code,
59 TfLiteStatus AsymmetricQuantizeTensor(BuiltinOperator op_code,
|
| D | subgraph_quantizer.cc | 56 bool OpHasOptionalBiasTensor(BuiltinOperator op_code) { in OpHasOptionalBiasTensor()
68 const OpWithBiasTensors* GetInfoForOpWithBiasTensor(BuiltinOperator op_code) { in GetInfoForOpWithBiasTensor()
197 BuiltinOperator op_code, int32_t tensor_idx) { in AsymmetricQuantizeTensor()
222 TfLiteStatus SubgraphQuantizer::QuantizeOpWithBias(BuiltinOperator op_code, in QuantizeOpWithBias()
268 BuiltinOperator op_code, OperatorT* op) { in PropagateMinMaxForAvgAndMaxPool()
311 BuiltinOperator op_code, OperatorT* op) { in AsymmetricQuantizeSoftmax()
332 BuiltinOperator op_code, OperatorT* op) { in AsymmetricQuantizeInputsAndOutputs()
361 const BuiltinOperator op_code = in QuantizeOperator()
|
| D | quantize_weights.cc | 72 std::vector<int32_t> GetWeightInputIndices(const BuiltinOperator& op_code) { in GetWeightInputIndices()
103 bool IsHybridEvaluationOp(const OperatorT* op, const BuiltinOperator& op_code) { in IsHybridEvaluationOp()
127 const BuiltinOperator& op_code) { in CheckAllOpInputsQuantized()
152 const BuiltinOperator op_code = in InsertQuantizableInputTensorsFromOperator()
236 const BuiltinOperator& op_code = model->operator_codes[i]->builtin_code; in UpdateInt8OperatorVersions()
296 const BuiltinOperator consumer_op_code = in QuantizeWeightsInternal()
|
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver.h | 33 struct ValueHasher<tflite::BuiltinOperator> {
34 size_t operator()(const tflite::BuiltinOperator& v) const {
57 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
60 void AddBuiltin(tflite::BuiltinOperator op,
68 typedef std::pair<tflite::BuiltinOperator, int> BuiltinOperatorKey;
|
| D | mutable_op_resolver.cc | 20 const TfLiteRegistration* MutableOpResolver::FindOp(tflite::BuiltinOperator op, in FindOp()
32 void MutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | micro_mutable_op_resolver.h | 29 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
32 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration,
|
| D | micro_mutable_op_resolver.cc | 21 tflite::BuiltinOperator op, int version) const { in FindOp()
45 void MicroMutableOpResolver::AddBuiltin(tflite::BuiltinOperator op, in AddBuiltin()
|
| D | micro_interpreter.cc | 46 return EnumNameBuiltinOperator(BuiltinOperator(registration->builtin_code)); in OpNameFromRegistration()
184 BuiltinOperator op_type = in Invoke()
185 static_cast<BuiltinOperator>(registration->builtin_code); in Invoke()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | logging_op_resolver.h | 42 const TfLiteRegistration* FindOp(BuiltinOperator op,
45 KernelEvalFuncPtr GetWrappedKernelInvoke(BuiltinOperator op,
|
| D | calibration_common.h | 26 using BuiltinOperatorKey = std::pair<BuiltinOperator, int>;
47 BuiltinOperator builtin_op_code;
|
| D | logging_op_resolver.cc | 38 const TfLiteRegistration* LoggingOpResolver::FindOp(BuiltinOperator op, in FindOp()
49 KernelEvalFuncPtr LoggingOpResolver::GetWrappedKernelInvoke(BuiltinOperator op, in GetWrappedKernelInvoke()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | maximum_minimum_test.cc | 28 MaxMinOpModel(tflite::BuiltinOperator op, const TensorData& input1, in MaxMinOpModel()
61 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
75 void TestModel(tflite::BuiltinOperator op, const TensorData& input1, in TestModel()
|
| D | logical_test.cc | 29 std::initializer_list<int> input2_shape, BuiltinOperator op) { in LogicalOpModel()
48 void ConfigureBuiltinOp(BuiltinOperator op) { in ConfigureBuiltinOp()
|
| D | elementwise_test.cc | 39 ElementWiseOpFloatModel(BuiltinOperator op, in ElementWiseOpFloatModel()
50 ElementWiseOpBoolModel(BuiltinOperator op, in ElementWiseOpBoolModel()
|
| D | register.h | 30 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
|
| D | register_ref.h | 30 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v0.fbs | 55 enum BuiltinOperator : byte {
204 // An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
207 builtin_code:BuiltinOperator;
|
| /external/tensorflow/tensorflow/lite/experimental/writer/ |
| D | writer_lib.cc | 34 FlatBufferBuilder* fbb, enum BuiltinOperator op, void* builtin_op_data) { in CreateBuiltinUnion()
84 fbb, static_cast<enum BuiltinOperator>(registration.builtin_code), in ExportOperators()
211 *fbb, static_cast<BuiltinOperator>(it.builtin), custom_name)); in CreateOpCodeTable()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | op_resolver.h | 30 virtual const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
|
| /external/tensorflow/tensorflow/lite/experimental/c/ |
| D | c_api_experimental.cc | 33 options->op_resolver.AddBuiltin(static_cast<tflite::BuiltinOperator>(op), in TFL_InterpreterOptionsAddBuiltinOp()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.h | 29 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op, in FindOp()
|