Home
last modified time | relevance | path

Searched refs:TfLiteArgMaxParams (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc70 TfLiteArgMaxParams arg_max_params; in TEST()
Dbuiltin_op_data.h351 } TfLiteArgMaxParams; typedef
/external/tensorflow/tensorflow/lite/kernels/
Darg_min_max.cc65 auto* params = reinterpret_cast<TfLiteArgMaxParams*>(node->builtin_data); in Prepare()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc615 auto params = safe_allocator.Allocate<TfLiteArgMaxParams>(); in ParseOpData()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1433 reinterpret_cast<TfLiteArgMaxParams*>(node->builtin_data); in Validate()