Home
last modified time | relevance | path

Searched refs:HasSoftmax (Results 1 – 3 of 3) sorted by relevance

/external/libtextclassifier/native/lang_id/common/
Dembedding-network-params.h144 bool HasSoftmax() const { in HasSoftmax() function
153 SAFTM_CHECK(HasSoftmax()) << "No softmax layer."; in GetSoftmaxMatrix()
168 SAFTM_CHECK(HasSoftmax()) << "No softmax layer."; in GetSoftmaxBias()
Dembedding-network.cc313 SAFTM_CHECK(model_->HasSoftmax()); in EmbeddingNetwork()
/external/libtextclassifier/native/lang_id/common/flatbuffers/
Dembedding-network-params-from-flatbuffer.cc237 if (HasSoftmax()) { in ValidityChecking()