Searched refs:hidden_size (Results 1 – 4 of 4) sorted by relevance
51 hidden_bias_blob_offset_ = hidden_blob_offset_ + hidden_size(); in EmbeddingNetworkParamsFromImage()114 int hidden_size() const override { return trimmed_proto_.hidden_size(); } in hidden_size() function117 TC_DCHECK(InRange(i, hidden_size())); in hidden_num_rows()122 TC_DCHECK(InRange(i, hidden_size())); in hidden_num_cols()127 TC_DCHECK(InRange(i, hidden_size())); in hidden_weights()
101 if (hidden_size() != hidden_bias_size()) { in GetNumHiddenLayers()102 TC_LOG(ERROR) << "Hidden layer mismatch " << hidden_size() in GetNumHiddenLayers()105 return std::max(0, std::min(hidden_size(), hidden_bias_size())); in GetNumHiddenLayers()113 TC_DCHECK(InRange(i, hidden_size())); in GetHiddenLayerMatrix()248 virtual int hidden_size() const = 0;
147 int hidden_size() const override { return proto_->hidden_size(); } in hidden_size() function150 TC_DCHECK(InRange(i, hidden_size())); in hidden_num_rows()155 TC_DCHECK(InRange(i, hidden_size())); in hidden_num_cols()160 TC_DCHECK(InRange(i, hidden_size())); in hidden_weights()
60 // must have hidden_bias_size() == hidden_size() + 1 (we store weights for