Searched refs:TfLiteSoftmaxParams (Results 1 – 12 of 12) sorted by relevance
39 const TfLiteSoftmaxParams* softmax_params = in PopulateSubGraph()40 reinterpret_cast<const TfLiteSoftmaxParams*>(builtin_data_); in PopulateSubGraph()
33 const TfLiteSoftmaxParams* params, in CalculateSoftmaxParams()141 auto* params = static_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxPrepare()
257 TfLiteSoftmaxParams builtin_data = {1.0f}; in ValidateSoftmaxGoldens()
41 TfLiteSoftmaxParams softmax_params; in TEST()
203 } TfLiteSoftmaxParams; typedef
119 const TfLiteSoftmaxParams* params, in CalculateSoftmaxOpDataHifimini()158 auto* params = static_cast<TfLiteSoftmaxParams*>(node->builtin_data); in PrepareHifimini()
582 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxPrepare()1051 TfLiteTensor* output, TfLiteSoftmaxParams* params) { in SoftmaxFloat()1124 auto* params = reinterpret_cast<TfLiteSoftmaxParams*>(node->builtin_data); in SoftmaxEval()
141 const auto* softmax_params = reinterpret_cast<const TfLiteSoftmaxParams*>(node->builtin_data);
1790 std::unique_ptr<TfLiteSoftmaxParams, in ParseSoftmax()1792 params = safe_allocator.Allocate<TfLiteSoftmaxParams>(); in ParseSoftmax()
981 const TfLiteSoftmaxParams* softmax_params = in VisitNode()982 static_cast<const TfLiteSoftmaxParams*>(node->builtin_data); in VisitNode()2713 const TfLiteSoftmaxParams* params, in VisitSoftmaxNode()
1751 const TfLiteSoftmaxParams* tf_options; in IsSupported()1768 const TfLiteSoftmaxParams* tf_options; in Parse()
2849 auto builtin = reinterpret_cast<TfLiteSoftmaxParams*>( in Map()