Searched refs:out_weights (Results 1 – 5 of 5) sorted by relevance
246 Tensor* out_weights = nullptr; in Compute() local250 2, TensorShape({static_cast<int>(weights.size())}), &out_weights)); in Compute()255 out_weights->vec<float>()(i) = weights[i]; in Compute()
85 out_weights = tf.Variable(107 prediction = tf.matmul(outputs[-1], out_weights) + out_bias
92 out_weights = tf.Variable(121 prediction = tf.matmul(output, out_weights) + out_bias
89 out_weights = tf.Variable(111 prediction = tf.matmul(outputs[-1], out_weights) + out_bias
96 out_weights = tf.Variable(142 prediction = tf.matmul(output, out_weights) + out_bias