Home
last modified time | relevance | path

Searched refs:softmax_bias_size (Results 1 – 3 of 3) sorted by relevance

/external/libtextclassifier/common/
Dembedding-network-params.h143 if (softmax_size() != softmax_bias_size()) { in HasSoftmaxLayer()
145 << " != " << softmax_bias_size(); in HasSoftmaxLayer()
147 return (softmax_size() == 1) && (softmax_bias_size() == 1); in HasSoftmaxLayer()
172 TC_DCHECK(softmax_bias_size() == 1); in GetSoftmaxBias()
292 virtual int softmax_bias_size() const = 0;
Dembedding-network-params-from-proto.h198 int softmax_bias_size() const override { in softmax_bias_size() function
203 TC_DCHECK(InRange(i, softmax_bias_size())); in softmax_bias_num_rows()
208 TC_DCHECK(InRange(i, softmax_bias_size())); in softmax_bias_num_cols()
213 TC_DCHECK(InRange(i, softmax_bias_size())); in softmax_bias_weights()
/external/libtextclassifier/common/memory_image/
Dembedding-network-params-from-image.h175 int softmax_bias_size() const override { in softmax_bias_size() function
180 TC_DCHECK(InRange(i, softmax_bias_size())); in softmax_bias_num_rows()
185 TC_DCHECK(InRange(i, softmax_bias_size())); in softmax_bias_num_cols()
190 TC_DCHECK(InRange(i, softmax_bias_size())); in softmax_bias_weights()