Lines Matching refs:network
23 #include "common/embedding-network.h"
25 #include "common/memory_image/embedding-network-params-from-image.h"
131 // NOTE: The network object needs to be available at the time when the returned
133 FeatureVectorFn CreateFeatureVectorFn(const EmbeddingNetwork& network,
136 return [&network, sparse_embedding_size, feature_type](
143 if (network.GetEmbedding(feature_vector, 0, embedding)) {
236 const FeatureProcessor& feature_processor, const EmbeddingNetwork& network,
242 const int embedding_size = network.EmbeddingSize(0);
245 CreateFeatureVectorFn(network, embedding_size),
268 network.ComputeLogits(features, &scores);
