Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/micro/kernels/
Dpooling.cc43 const TfLitePoolParams* params, in CalculateOpData()
62 const TfLitePoolParams* params, const OpData* data, in AverageEvalFloat()
80 const TfLitePoolParams* params, const OpData* data, in AverageEvalQuantized()
110 TfLitePoolParams* params, const OpData* data, in MaxEvalFloat()
128 TfLitePoolParams* params, const OpData* data, in MaxEvalQuantized()
157 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in AverageEval()
186 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in MaxEval()
219 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in Prepare()
Dpooling_test.cc41 TfLitePoolParams builtin_data = {padding, in ValidatePoolingGoldens()
/external/tensorflow/tensorflow/lite/kernels/
Dpooling.cc69 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in GenericPrepare()
117 TfLitePoolParams* params, OpData* data, in AverageEvalFloat()
145 TfLitePoolParams* params, OpData* data, in AverageEvalQuantizedUint8()
175 TfLitePoolParams* params, OpData* data, in AverageEvalQuantizedInt8()
205 TfLitePoolParams* params, OpData* data, in AverageEvalQuantizedInt16()
231 TfLitePoolParams* params, OpData* data, in MaxEvalFloat()
258 TfLitePoolParams* params, OpData* data, in MaxEvalQuantizedUInt8()
287 TfLitePoolParams* params, OpData* data, in MaxEvalQuantizedInt8()
316 TfLitePoolParams* params, OpData* data, in MaxEvalQuantizedInt16()
341 TfLitePoolParams* params, OpData* data, in L2EvalFloat()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dpooling.cc46 const TfLitePoolParams* params, in CalculateOpData()
75 const TfLitePoolParams* params, const OpData& data, in AverageEvalFloat()
97 const TfLitePoolParams* params, const OpData& data, in AverageEvalQuantized()
169 TfLitePoolParams* params, const OpData& data, in MaxEvalFloat()
190 TfLitePoolParams* params, const OpData& data, in MaxEvalQuantizedUInt8()
209 const TfLitePoolParams* params, const OpData& data, in MaxEvalInt8()
271 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in MaxPrepare()
286 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in AveragePrepare()
317 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in AverageEval()
344 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in MaxEval()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dpooling.cc58 const TfLitePoolParams* params) { in IsMliApplicable()
64 const TfLitePoolParams* params, in CalculateOpData()
88 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in Prepare()
145 const TfLitePoolParams* params, const OpData& data, in AverageEvalFloat()
173 TfLiteStatus EvalMli(TfLiteContext* context, const TfLitePoolParams* params, in EvalMli()
238 const TfLitePoolParams* params, const OpData& data, in AverageEvalQuantized()
274 TfLitePoolParams* params, const OpData& data, in MaxEvalFloat()
299 TfLitePoolParams* params, const OpData& data, in MaxEvalQuantized()
335 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in AverageEval()
367 auto* params = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in MaxEval()
Dpooling_slicing_test.cc68 TfLitePoolParams builtin_data = {padding, stride_width, stride_height, in TestAveragePoolingQuantized()
139 TfLitePoolParams builtin_data = { in TestMaxPoolQuantized()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dpool_2d_builder.cc38 const TfLitePoolParams* pool_params = in PopulateSubGraph()
39 reinterpret_cast<const TfLitePoolParams*>(builtin_data_); in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dpooling_layer_builder.cc60 const TfLitePoolParams* params = in Build()
61 reinterpret_cast<const TfLitePoolParams*>(builtin_data_); in Build()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc237 const TfLitePoolParams* pool_params = in IsNodeSupportedByHexagon()
238 reinterpret_cast<const TfLitePoolParams*>(node->builtin_data); in IsNodeSupportedByHexagon()
244 const TfLitePoolParams* pool_params = in IsNodeSupportedByHexagon()
245 reinterpret_cast<const TfLitePoolParams*>(node->builtin_data); in IsNodeSupportedByHexagon()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils.cc107 TfLitePoolParams pool_params; in CreateCustomOptions()
143 sizeof(TfLitePoolParams)); in CreateCustomOptions()
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_unpooling_2d_test.cc42 TfLitePoolParams params{padding, stride_width, stride_height, in MaxUnpoolingOpModel()
46 custom_option.assign(params_ptr, params_ptr + sizeof(TfLitePoolParams)); in MaxUnpoolingOpModel()
Dmax_unpooling_2d.cc57 reinterpret_cast<const TfLitePoolParams*>(node->custom_initial_data); in Prepare()
Dmax_pool_with_argmax.cc108 TfLitePoolParams params;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperation_parser.h61 absl::Status ParsePoolingAttributes(const TfLitePoolParams* tf_options,
Doperation_parser.cc77 absl::Status ParsePoolingAttributes(const TfLitePoolParams* tf_options, in ParsePoolingAttributes()
Dmodel_builder.cc1335 const TfLitePoolParams* tf_options; in IsSupported()
1362 const TfLitePoolParams* tf_options; in Parse()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc31 TfLitePoolParams pool_params; in TEST()
Dbuiltin_op_data.h111 } TfLitePoolParams; typedef
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc474 const TfLitePoolParams* params, in CheckMediaPipePoolParams()
566 const TfLitePoolParams* params, in CheckPoolingParams()
835 const TfLitePoolParams* pool_params = in VisitNode()
836 static_cast<const TfLitePoolParams*>(node->builtin_data); in VisitNode()
914 const TfLitePoolParams* pool_params = in VisitNode()
915 static_cast<const TfLitePoolParams*>(node->builtin_data); in VisitNode()
1017 TfLitePoolParams pool_params = {kTfLitePaddingUnknown}; in VisitNode()
1025 TfLitePoolParams pool_params = {kTfLitePaddingUnknown}; in VisitNode()
1126 const TfLitePoolParams* pool_params, in VisitAveragePool2DNode()
1841 const TfLitePoolParams* pool_params, in VisitMaxPool2DNode()
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/
Dcoreml_delegate.mm84 const auto* params = reinterpret_cast<const TfLitePoolParams*>(node->builtin_data);
106 const auto* params = reinterpret_cast<const TfLitePoolParams*>(node->builtin_data);
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc1555 std::unique_ptr<TfLitePoolParams, in ParsePool()
1557 params = safe_allocator.Allocate<TfLitePoolParams>(); in ParsePool()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc406 static Optional<TfLitePoolParams> GetTflitePoolParams(Operation* inst, in GetTflitePoolParams()
408 TfLitePoolParams pool_params; in GetTflitePoolParams()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc774 auto builtin = reinterpret_cast<TfLitePoolParams*>(data); in AddPoolingParams()
1758 auto builtin = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in Validate()
1778 auto builtin = reinterpret_cast<TfLitePoolParams*>(node->builtin_data); in Validate()