Searched refs:num_hidden_layers (Results 1 – 1 of 1) sorted by relevance
340 const int num_hidden_layers = model->GetNumHiddenLayers(); in EmbeddingNetwork() local341 if (num_hidden_layers < 1) { in EmbeddingNetwork()342 TC_LOG(ERROR) << "Wrong number of hidden layers: " << num_hidden_layers; in EmbeddingNetwork()345 hidden_weights_.resize(num_hidden_layers); in EmbeddingNetwork()346 hidden_bias_.resize(num_hidden_layers); in EmbeddingNetwork()348 for (int i = 0; i < num_hidden_layers; ++i) { in EmbeddingNetwork()