Searched refs:softmax_bias (Results 1 – 3 of 3) sorted by relevance
| /external/libtextclassifier/native/lang_id/common/ |
| D | embedding-network.cc | 316 EmbeddingNetworkParams::Matrix softmax_bias = model_->GetSoftmaxBias(); in EmbeddingNetwork() local
317 SAFTM_CHECK_EQ(1, softmax_bias.cols); in EmbeddingNetwork()
318 CheckNoQuantization(softmax_bias); in EmbeddingNetwork()
319 layer_bias_.push_back(softmax_bias); in EmbeddingNetwork()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker_v2_test.py | 326 softmax_bias = constant_op.constant(
333 all_params = [inp, hidden_weight, hidden_bias, softmax_weight, softmax_bias]
336 def f(inp, hidden_weight, hidden_bias, softmax_weight, softmax_bias): argument
340 features, softmax_weight, softmax_bias, name="logits")
|
| /external/tensorflow/tensorflow/python/ops/v1_compat_tests/ |
| D | gradient_checker_test.py | 250 softmax_bias = constant_op.constant(
258 inp, hidden_weight, hidden_bias, softmax_weight, softmax_bias
272 features, softmax_weight, softmax_bias, name="logits")
|