Searched refs:SafeGetNumLayers (Results 1 – 2 of 2) sorted by relevance
89 return std::max(SafeGetNumLayers() - 1, 0); in hidden_size()129 int softmax_size() const override { return (SafeGetNumLayers() > 0) ? 1 : 0; } in softmax_size()208 int SafeGetNumLayers() const;
307 int EmbeddingNetworkParamsFromFlatbuffer::SafeGetNumLayers() const { in SafeGetNumLayers() function in libtextclassifier3::mobile::EmbeddingNetworkParamsFromFlatbuffer318 if (!InRangeIndex(i, SafeGetNumLayers(), "layer")) { in SafeGetLayer()394 int num_layers = SafeGetNumLayers(); in SafeGetSoftmaxLayer()