Searched refs:softmax_bias (Results 1 – 4 of 4) sorted by relevance
204 return proto_->has_softmax_bias() ? proto_->softmax_bias().rows() : 0; in softmax_bias_num_rows()209 return proto_->has_softmax_bias() ? proto_->softmax_bias().cols() : 0; in softmax_bias_num_cols()214 return proto_->has_softmax_bias() ? proto_->softmax_bias().value().data() in softmax_bias_weights()
364 const EmbeddingNetworkParams::Matrix softmax_bias = model->GetSoftmaxBias(); in EmbeddingNetwork() local366 !InitNonQuantizedVector(softmax_bias, &softmax_bias_)) { in EmbeddingNetwork()
68 optional MatrixParams softmax_bias = 5; field
181 return trimmed_proto_.softmax_bias().rows(); in softmax_bias_num_rows()186 return trimmed_proto_.softmax_bias().cols(); in softmax_bias_num_cols()