| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| lstm_ops.py | 54 use_peephole=None,
68 if not use_peephole:
101 use_peephole: An optional `bool`. Defaults to `False`.
138 use_peephole=use_peephole,
154 use_peephole=None,
170 use_peephole: An optional `bool`. Defaults to `False`.
226 use_peephole=use_peephole)
273 use_peephole=op.get_attr("use_peephole")
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| lstm_ops_test.py | 107 def blocks_match(sess, use_peephole, dtype=dtypes.float32, cell_clip=None):
125 if use_peephole:
145 use_peepholes=use_peephole,
153 if use_peephole:
163 use_peephole=True)
175 use_peephole=use_peephole,
184 if use_peephole:
193 if use_peephole:
277 pcell = lstm_ops.LSTMBlockCell(10, use_peephole=True
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| lstm_ops_gpu.cu.cc | 84 // TODO(b/67600500): Try making 'use_peephole' a template parameter.
85 template <typename T, bool use_peephole>
149 if (use_peephole) {
162 if (use_peephole) {
181 if (use_peephole) {
224 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x,
264 if (use_peephole) {
303 const int batch_size, const int cell_size, const bool use_peephole) {
328 if (use_peephole) {
351 if (use_peephole) {
[all...] |
| lstm_ops.h | 158 bool use_peephole, typename TTypes<T>::ConstMatrix x,
182 OpKernelContext* ctx, const Device& d, bool use_peephole,
207 OpKernelContext* ctx, const Device& d, bool use_peephole,
235 if (use_peephole) {
255 if (use_peephole) {
284 if (use_peephole) {
|
| lstm_ops.cc | 47 const float forget_bias, const float cell_clip, bool use_peephole,
74 if (use_peephole) {
88 if (use_peephole) {
111 if (use_peephole) {
128 bool use_peephole, typename TTypes<T>::ConstMatrix x,
151 if (use_peephole) {
171 if (use_peephole) {
185 const float cell_clip, bool use_peephole, \
197 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
202 OpKernelContext* ctx, const CPUDevice& d, bool use_peephole, \
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/ |
| lstm_eval.cc | 124 const bool use_peephole = (cell_to_output_weights_ptr != nullptr); local
207 if (use_peephole) {
227 if (use_peephole) {
274 if (use_peephole) {
478 const bool use_peephole = (cell_to_output_weights_ptr != nullptr); local
658 if (use_peephole && !is_cell_state_all_zeros) {
681 if (use_peephole && !is_cell_state_all_zeros) {
918 const bool use_peephole = (cell_to_output_weights != nullptr); local
1121 const bool use_peephole = (cell_to_output_weights != nullptr); local
[all...] |
| optional_tensor_test.cc | 34 bool use_peephole, bool use_projection_weights,
63 if (use_peephole) {
238 /*use_cifg=*/true, /*use_peephole=*/true,
|
| lstm_test.cc | 38 bool use_peephole, bool use_projection_weights,
69 if (use_peephole) {
272 bool use_cifg, bool use_peephole,
277 : LSTMOpModel(n_batch, n_input, n_cell, n_output, use_cifg, use_peephole,
457 /*use_cifg=*/false, /*use_peephole=*/false,
515 /*use_cifg=*/false, /*use_peephole=*/false,
574 /*use_cifg=*/false, /*use_peephole=*/false,
678 /*use_cifg=*/true, /*use_peephole=*/true,
736 /*use_cifg=*/true, /*use_peephole=*/true,
795 /*use_cifg=*/true, /*use_peephole=*/true
[all...] |
| unidirectional_sequence_lstm_test.cc | 36 bool use_peephole, bool use_projection_weights,
68 if (use_peephole) {
244 bool time_major, bool use_cifg, bool use_peephole,
250 use_cifg, use_peephole, use_projection_weights, use_projection_bias,
460 /*time_major=*/true, /*use_cifg=*/false, /*use_peephole=*/false,
522 /*time_major=*/true, /*use_cifg=*/false, /*use_peephole=*/false,
589 /*time_major=*/true, /*use_cifg=*/false, /*use_peephole=*/false,
652 /*time_major=*/true, /*use_cifg=*/false, /*use_peephole=*/false,
761 /*time_major=*/true, /*use_cifg=*/true, /*use_peephole=*/true,
823 /*time_major=*/true, /*use_cifg=*/true, /*use_peephole=*/true
[all...] |
| bidirectional_sequence_lstm_test.cc | 39 bool use_peephole, bool use_projection_weights,
76 if (use_peephole) {
131 if (use_peephole) {
416 /*use_peephole=*/false, /*use_projection_weights=*/false,
581 /*use_peephole=*/false, /*use_projection_weights=*/false,
745 /*use_peephole=*/false, /*use_projection_weights=*/false,
[all...] |
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| nnapi_delegate_test.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | 71 num_units, forget_bias=0, cell_clip=None, use_peephole=False,
[all...] |
