Searched refs:saft_fbs (Results 1 – 8 of 8) sorted by relevance
63 const saft_fbs::Matrix *matrix = SafeGetEmbeddingMatrix(i); in embeddings_num_rows()68 const saft_fbs::Matrix *matrix = SafeGetEmbeddingMatrix(i); in embeddings_num_cols()73 const saft_fbs::Matrix *matrix = SafeGetEmbeddingMatrix(i); in embeddings_weights()78 const saft_fbs::Matrix *matrix = SafeGetEmbeddingMatrix(i); in embeddings_quant_type()83 const saft_fbs::Matrix *matrix = SafeGetEmbeddingMatrix(i); in embeddings_quant_scales()93 const saft_fbs::Matrix *weights = SafeGetLayerWeights(i); in hidden_num_rows()98 const saft_fbs::Matrix *weights = SafeGetLayerWeights(i); in hidden_num_cols()103 const saft_fbs::Matrix *weights = SafeGetLayerWeights(i); in hidden_weights_quant_type()108 const saft_fbs::Matrix *weights = SafeGetLayerWeights(i); in hidden_weights()115 const saft_fbs::Matrix *bias = SafeGetLayerBias(i); in hidden_bias_num_rows()[all …]
193 if (!saft_fbs::VerifyEmbeddingNetworkBuffer(verifier)) { in EmbeddingNetworkParamsFromFlatbuffer()197 network_ = saft_fbs::GetEmbeddingNetwork(start); in EmbeddingNetworkParamsFromFlatbuffer()277 const saft_fbs::InputChunk *289 const saft_fbs::InputChunk *input_chunk = input_chunks->Get(i); in SafeGetInputChunk()296 const saft_fbs::Matrix *298 const saft_fbs::InputChunk *input_chunk = SafeGetInputChunk(i); in SafeGetEmbeddingMatrix()300 const saft_fbs::Matrix *matrix = input_chunk->embedding(); in SafeGetEmbeddingMatrix()316 const saft_fbs::NeuralLayer *EmbeddingNetworkParamsFromFlatbuffer::SafeGetLayer( in SafeGetLayer()328 const saft_fbs::NeuralLayer *layer = layers->Get(i); in SafeGetLayer()335 const saft_fbs::Matrix *[all …]
30 namespace saft_fbs {
27 namespace saft_fbs { namespace
23 namespace libtextclassifier3.saft_fbs;
21 namespace libtextclassifier3.saft_fbs;
65 model_ = saft_fbs::GetVerifiedModelFromBytes(model_bytes); in Initialize()72 if (!saft_fbs::FillParameters(*model_, &context_)) { in Initialize()101 saft_fbs::GetInputBytes(saft_fbs::GetInputByName(model_, kInputName)); in InitNetworkParams()
102 const saft_fbs::Model *model_;