Searched refs:softmax_params (Results 1 – 4 of 4) sorted by relevance
39 const TfLiteSoftmaxParams* softmax_params = in PopulateSubGraph() local41 beta_value_ = softmax_params->beta; in PopulateSubGraph()
41 TfLiteSoftmaxParams softmax_params; in TEST() local
141 const auto* softmax_params = reinterpret_cast<const TfLiteSoftmaxParams*>(node->builtin_data);142 return softmax_params != nullptr && softmax_params->beta == 1.0;
981 const TfLiteSoftmaxParams* softmax_params = in VisitNode() local985 context->tensors, softmax_params, in VisitNode()