Searched refs:TfLiteSoftmaxParams (Results 1 – 9 of 9) sorted by relevance
41 const TfLiteSoftmaxParams* params, in CalculateSoftmaxOpData()72 TfLiteSoftmaxParams* params) { in Softmax1DFloat()80 TfLiteSoftmaxParams* params) { in Softmax2DFloat()88 TfLiteSoftmaxParams* params, OpData* data) { in Softmax1DQuantized()106 TfLiteSoftmaxParams* params, OpData* data) { in Softmax2DQuantized()126 TfLiteSoftmaxParams* params) { in Softmax4DFloat()135 TfLiteSoftmaxParams* params, OpData* data) { in Softmax4DQuantized()146 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxEval()
52 TfLiteSoftmaxParams builtin_data = {1.0f}; in TestSoftmaxFloat()118 TfLiteSoftmaxParams builtin_data = {1.0f}; in TestSoftmaxQuantized()
236 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxPrepare()582 TfLiteSoftmaxParams* params) { in Softmax1DFloat()589 TfLiteSoftmaxParams* params) { in Softmax2DFloat()597 TfLiteSoftmaxParams* params) { in Softmax3DFloat()611 TfLiteSoftmaxParams* params, OpData* data) { in Softmax1DQuantizedUint8()627 TfLiteSoftmaxParams* params, OpData* data) { in Softmax2DQuantizedUint8()646 TfLiteSoftmaxParams* params, OpData* data) { in Softmax3DQuantizedUint8()663 TfLiteSoftmaxParams* params) { in Softmax4DFloat()672 TfLiteSoftmaxParams* params, OpData* data) { in Softmax4DQuantizedUint8()685 TfLiteSoftmaxParams* params, OpData* data) { in Softmax1DQuantizedInt8()[all …]
41 TfLiteSoftmaxParams softmax_params; in TEST()
148 } TfLiteSoftmaxParams; typedef
307 TfLiteSoftmaxParams* params = in ParseOpData()308 allocator->AllocatePOD<TfLiteSoftmaxParams>(); in ParseOpData()
377 auto builtin = reinterpret_cast<TfLiteSoftmaxParams*>(data); in AddOpsAndParams()
347 auto builtin = reinterpret_cast<TfLiteSoftmaxParams*>(data); in AddOpsAndParams()
646 auto builtin = reinterpret_cast<TfLiteSoftmaxParams*>( in Map()