Searched refs:softmax_size (Results 1 – 3 of 3) sorted by relevance
143 if (softmax_size() != softmax_bias_size()) { in HasSoftmaxLayer()144 TC_LOG(ERROR) << "Softmax layer mismatch " << softmax_size() in HasSoftmaxLayer()147 return (softmax_size() == 1) && (softmax_bias_size() == 1); in HasSoftmaxLayer()155 TC_DCHECK(softmax_size() == 1); in GetSoftmaxMatrix()277 virtual int softmax_size() const = 0;
181 int softmax_size() const override { return proto_->has_softmax() ? 1 : 0; } in softmax_size() function184 TC_DCHECK(InRange(i, softmax_size())); in softmax_num_rows()189 TC_DCHECK(InRange(i, softmax_size())); in softmax_num_cols()194 TC_DCHECK(InRange(i, softmax_size())); in softmax_weights()
53 softmax_bias_blob_offset_ = softmax_blob_offset_ + softmax_size(); in EmbeddingNetworkParamsFromImage()154 int softmax_size() const override { in softmax_size() function159 TC_DCHECK(InRange(i, softmax_size())); in softmax_num_rows()164 TC_DCHECK(InRange(i, softmax_size())); in softmax_num_cols()169 TC_DCHECK(InRange(i, softmax_size())); in softmax_weights()