HomeSort by relevance Sort by last modified time
    Searched refs:softmax_bias (Results 1 - 3 of 3) sorted by null

  /external/libtextclassifier/lang_id/common/
embedding-network.cc 316 EmbeddingNetworkParams::Matrix softmax_bias = model_->GetSoftmaxBias(); local
317 SAFTM_CHECK_EQ(1, softmax_bias.cols);
318 CheckNoQuantization(softmax_bias);
319 layer_bias_.push_back(softmax_bias);
  /external/tensorflow/tensorflow/python/ops/
gradient_checker_v2_test.py 244 softmax_bias = constant_op.constant(
248 name="softmax_bias")
252 inp, hidden_weight, hidden_bias, softmax_weight, softmax_bias
256 def f(inp, hidden_weight, hidden_bias, softmax_weight, softmax_bias):
260 features, softmax_weight, softmax_bias, name="logits")
296 self.assertLess(self._BuildAndTestMiniMNIST(4, "softmax_bias"), 1e-8)
gradient_checker_test.py 250 softmax_bias = constant_op.constant(
254 name="softmax_bias")
258 inp, hidden_weight, hidden_bias, softmax_weight, softmax_bias
266 ] # softmax_bias
272 features, softmax_weight, softmax_bias, name="logits")
309 self.assertLess(self._BuildAndTestMiniMNIST(4, "softmax_bias"), 1e-8)

Completed in 371 milliseconds