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.h403 } TfLiteArgMaxParams; typedef
/external/tensorflow/tensorflow/lite/kernels/
Darg_min_max.cc81 auto* params = reinterpret_cast<TfLiteArgMaxParams*>(node->builtin_data); in Prepare()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc938 std::unique_ptr<TfLiteArgMaxParams, in ParseArgMax()
940 params = safe_allocator.Allocate<TfLiteArgMaxParams>(); in ParseArgMax()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1729 reinterpret_cast<TfLiteArgMaxParams*>(node->builtin_data); in Validate()