Home
last modified time | relevance | path

Searched refs:TfLiteGatherParams (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dgather.cc40 reinterpret_cast<const TfLiteGatherParams*>(node->builtin_data); in Prepare()
107 TfLiteStatus Gather(const TfLiteGatherParams& params, const TfLiteTensor* input, in Gather()
139 reinterpret_cast<const TfLiteGatherParams*>(node->builtin_data); in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dgather.cc40 reinterpret_cast<const TfLiteGatherParams*>(node->builtin_data); in Prepare()
107 TfLiteStatus Gather(const TfLiteGatherParams& params, const TfLiteTensor* input, in Gather()
139 reinterpret_cast<const TfLiteGatherParams*>(node->builtin_data); in Eval()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc63 TfLiteGatherParams gather_params; in TEST()
Dbuiltin_op_data.h368 } TfLiteGatherParams; typedef
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc1311 auto params = safe_allocator.Allocate<TfLiteGatherParams>(); in ParseGather()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc3508 auto builtin = reinterpret_cast<TfLiteGatherParams*>( in Map()