Home
last modified time | relevance | path

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

/external/libtextclassifier/native/lang_id/common/
Dembedding-network-params.h152 Matrix GetSoftmaxMatrix() const { in GetSoftmaxMatrix() function
Dembedding-network.cc314 layer_weights_.push_back(model_->GetSoftmaxMatrix()); in EmbeddingNetwork()
/external/libtextclassifier/native/lang_id/common/flatbuffers/
Dembedding-network-params-from-flatbuffer.cc238 Matrix weights = GetSoftmaxMatrix(); in ValidityChecking()