Searched refs:hidden_bias_size (Results 1 – 4 of 4) sorted by relevance
52 softmax_blob_offset_ = hidden_bias_blob_offset_ + hidden_bias_size(); in EmbeddingNetworkParamsFromImage()133 int hidden_bias_size() const override { in hidden_bias_size() function134 return trimmed_proto_.hidden_bias_size(); in hidden_bias_size()138 TC_DCHECK(InRange(i, hidden_bias_size())); in hidden_bias_num_rows()143 TC_DCHECK(InRange(i, hidden_bias_size())); in hidden_bias_num_cols()148 TC_DCHECK(InRange(i, hidden_bias_size())); in hidden_bias_weights()
101 if (hidden_size() != hidden_bias_size()) { in GetNumHiddenLayers()103 << " != " << hidden_bias_size(); in GetNumHiddenLayers()105 return std::max(0, std::min(hidden_size(), hidden_bias_size())); in GetNumHiddenLayers()130 TC_DCHECK(InRange(i, hidden_bias_size())); in GetHiddenLayerBias()263 virtual int hidden_bias_size() const = 0;
164 int hidden_bias_size() const override { return proto_->hidden_bias_size(); } in hidden_bias_size() function167 TC_DCHECK(InRange(i, hidden_bias_size())); in hidden_bias_num_rows()172 TC_DCHECK(InRange(i, hidden_bias_size())); in hidden_bias_num_cols()177 TC_DCHECK(InRange(i, hidden_bias_size())); in hidden_bias_weights()
60 // must have hidden_bias_size() == hidden_size() + 1 (we store weights for