Searched refs:TfLiteSoftmaxParams (Results 1 – 10 of 10) sorted by relevance
43 const TfLiteSoftmaxParams* params, in CalculateSoftmaxOpData()85 TfLiteSoftmaxParams* params) { in Softmax1DFloat()93 TfLiteSoftmaxParams* params) { in Softmax2DFloat()101 TfLiteSoftmaxParams* params, OpData* data) { in Softmax1DQuantized()131 TfLiteSoftmaxParams* params, OpData* data) { in Softmax2DQuantized()163 TfLiteSoftmaxParams* params) { in Softmax4DFloat()172 TfLiteSoftmaxParams* params, OpData* data) { in Softmax4DQuantized()195 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxEval()
51 TfLiteSoftmaxParams builtin_data = {1.0f}; in TestSoftmaxFloat()117 TfLiteSoftmaxParams builtin_data = {1.0f}; in TestSoftmaxQuantized()183 TfLiteSoftmaxParams builtin_data = {1.0f}; in TestSoftmaxQuantizedSigned()
134 const TfLiteSoftmaxParams* params, in CalculateSoftmaxOpData()165 TfLiteSoftmaxParams* params, OpData* data) { in Softmax2DQuantized()192 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxPrepare()209 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxEval()
50 const TfLiteSoftmaxParams* softmax_params = in PopulateSubGraph()51 reinterpret_cast<const TfLiteSoftmaxParams*>(builtin_data_); in PopulateSubGraph()
41 TfLiteSoftmaxParams softmax_params; in TEST()
175 } TfLiteSoftmaxParams; typedef
546 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxPrepare()925 TfLiteTensor* output, TfLiteSoftmaxParams* params) { in SoftmaxFloat()964 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxEval()
343 auto params = safe_allocator.Allocate<TfLiteSoftmaxParams>(); in ParseOpData()
1793 TfLiteSoftmaxParams* tf_options = nullptr; in IsSupported()1811 reinterpret_cast<const TfLiteSoftmaxParams*>(tflite_node->builtin_data); in Parse()
2374 auto builtin = reinterpret_cast<TfLiteSoftmaxParams*>( in Map()