Searched refs:SafeGetSoftmaxLayer (Results 1 – 2 of 2) sorted by relevance
245 const saft_fbs::NeuralLayer *SafeGetSoftmaxLayer() const;248 const saft_fbs::NeuralLayer *layer = SafeGetSoftmaxLayer(); in SafeGetSoftmaxWeights()253 const saft_fbs::NeuralLayer *layer = SafeGetSoftmaxLayer(); in SafeGetSoftmaxBias()
393 EmbeddingNetworkParamsFromFlatbuffer::SafeGetSoftmaxLayer() const { in SafeGetSoftmaxLayer() function in libtextclassifier3::mobile::EmbeddingNetworkParamsFromFlatbuffer