Searched refs:out_weights (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
| D | k_feature_gradient_op.cc | 121 Tensor* out_weights = nullptr; in Compute() local
132 context, context->allocate_output(2, out_weights_shape, &out_weights)); in Compute()
144 auto weights_grad = out_weights->tensor<float, 3>(); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | candidate_sampler_ops.cc | 246 Tensor* out_weights = nullptr; in Compute() local
250 2, TensorShape({static_cast<int>(weights.size())}), &out_weights)); in Compute()
255 out_weights->vec<float>()(i) = weights[i]; in Compute()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | unidirectional_sequence_rnn_test.py | 70 out_weights = tf.Variable(
91 prediction = tf.matmul(outputs[-1], out_weights) + out_bias
|
| D | unidirectional_sequence_lstm_test.py | 74 out_weights = tf.Variable(
95 prediction = tf.matmul(outputs[-1], out_weights) + out_bias
|
| D | bidirectional_sequence_lstm_test.py | 75 out_weights = tf.Variable(
104 prediction = tf.matmul(output, out_weights) + out_bias
|
| D | bidirectional_sequence_rnn_test.py | 77 out_weights = tf.Variable(
123 prediction = tf.matmul(output, out_weights) + out_bias
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/g3doc/ |
| D | README.md | 78 out_weights = tf.Variable(tf.random_normal([self.units, self.num_class]))
190 out_weights = tf.Variable(tf.random_normal([self.units, self.num_class]))
206 logits = tf.matmul(outputs[-1], out_weights) + out_bias
|