Home
last modified time | relevance | path

Searched refs:TfLiteReducerParams (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dreduce_test.cc72 TfLiteReducerParams* params, in ValidateReduceGoldens()
96 TfLiteReducerParams* params, float tolerance = 1e-5) { in TestMeanFloatInput4D()
125 TfLiteReducerParams* params, float tolerance = 1e-5) { in TestReduceOpFloat()
154 const TfLiteRegistration& registration, TfLiteReducerParams* params) { in TestReduceOpQuantized()
192 TfLiteReducerParams* params) { in TestMeanOpQuantized()
231 TfLiteReducerParams params = {true}; in TF_LITE_MICRO_TEST()
250 TfLiteReducerParams params = { in TF_LITE_MICRO_TEST()
273 TfLiteReducerParams params = { in TF_LITE_MICRO_TEST()
289 TfLiteReducerParams params = {true}; in TF_LITE_MICRO_TEST()
308 TfLiteReducerParams params = { in TF_LITE_MICRO_TEST()
[all …]
Dreduce.cc143 TfLiteReducerParams* params = in EvalMean()
144 reinterpret_cast<TfLiteReducerParams*>(node->builtin_data); in EvalMean()
266 TfLiteReducerParams* params = in EvalMax()
267 static_cast<TfLiteReducerParams*>(node->builtin_data); in EvalMax()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc65 TfLiteReducerParams reducer_params; in TEST()
Dbuiltin_op_data.h376 } TfLiteReducerParams; typedef
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dpooling_layer_builder.cc141 reinterpret_cast<TfLiteReducerParams*>(node->builtin_data); in IsMeanOpSupported()
/external/tensorflow/tensorflow/lite/kernels/
Dreduce.cc58 params = reinterpret_cast<TfLiteReducerParams*>(node->builtin_data); in OpContext()
63 TfLiteReducerParams* params;
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc3400 auto builtin = reinterpret_cast<TfLiteReducerParams*>( in Map()
3566 auto builtin = reinterpret_cast<TfLiteReducerParams*>( in Map()
3572 auto builtin = reinterpret_cast<TfLiteReducerParams*>( in Map()
3578 auto builtin = reinterpret_cast<TfLiteReducerParams*>( in Map()
3590 auto builtin = reinterpret_cast<TfLiteReducerParams*>( in Map()
3596 auto builtin = reinterpret_cast<TfLiteReducerParams*>( in Map()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc1611 std::unique_ptr<TfLiteReducerParams, in ParseReducer()
1613 params = safe_allocator.Allocate<TfLiteReducerParams>(); in ParseReducer()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc925 const TfLiteReducerParams* reducer_params = in VisitNode()
926 static_cast<const TfLiteReducerParams*>(node->builtin_data); in VisitNode()
1947 const TfLiteReducerParams* reducer_params, in VisitMeanNode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc1403 const TfLiteReducerParams* tf_options; in Parse()