| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | lstm_ops.py | 54 use_peephole=None, argument
138 use_peephole=use_peephole,
154 use_peephole=None, argument
226 use_peephole=use_peephole)
273 use_peephole=op.get_attr("use_peephole"))
327 use_peephole=op.get_attr("use_peephole"))
352 use_peephole=False, argument
378 self._use_peephole = use_peephole
440 use_peephole=self._use_peephole)
604 use_peephole=False, argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops_gpu.cu.cc | 85 template <typename T, bool use_peephole>
149 if (use_peephole) { in lstm_gates()
162 if (use_peephole) { in lstm_gates()
181 if (use_peephole) { in lstm_gates()
224 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA() argument
264 if (use_peephole) { in LSTMBlockCellFpropWithCUDA()
303 const int batch_size, const int cell_size, const bool use_peephole) { in lstm_gates_bprop() argument
328 if (use_peephole) { in lstm_gates_bprop()
351 if (use_peephole) { in lstm_gates_bprop()
373 const bool use_peephole) { in LSTMBlockCellBpropWithCUDA() argument
[all …]
|
| D | 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, in operator()
235 if (use_peephole) { in operator()
255 if (use_peephole) { in operator()
284 if (use_peephole) { in operator()
|
| D | lstm_ops.cc | 47 const float forget_bias, const float cell_clip, bool use_peephole, in LSTMBlockCellFpropWithEigen() argument
74 if (use_peephole) { in LSTMBlockCellFpropWithEigen()
88 if (use_peephole) { in LSTMBlockCellFpropWithEigen()
111 if (use_peephole) { in LSTMBlockCellFpropWithEigen()
128 bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellBpropWithEigen() argument
151 if (use_peephole) { in LSTMBlockCellBpropWithEigen()
171 if (use_peephole) { in LSTMBlockCellBpropWithEigen()
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/contrib/rnn/python/kernel_tests/ |
| D | lstm_ops_test.py | 107 def blocks_match(sess, use_peephole, dtype=dtypes.float32, cell_clip=None): argument
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)
288 pcell = lstm_ops.LSTMBlockFusedCell(10, use_peephole=True)
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 124 const bool use_peephole = (cell_to_output_weights_ptr != nullptr); in LstmStepWithAuxInput() local
207 if (use_peephole) { in LstmStepWithAuxInput()
227 if (use_peephole) { in LstmStepWithAuxInput()
274 if (use_peephole) { in LstmStepWithAuxInput()
478 const bool use_peephole = (cell_to_output_weights_ptr != nullptr); in LstmStepWithAuxInput() local
658 if (use_peephole && !is_cell_state_all_zeros) { in LstmStepWithAuxInput()
681 if (use_peephole && !is_cell_state_all_zeros) { in LstmStepWithAuxInput()
733 if (use_peephole && !is_cell_state_all_zeros) { in LstmStepWithAuxInput()
918 const bool use_peephole = (cell_to_output_weights != nullptr); in EvalFloat() local
945 (use_peephole && !use_cifg) ? cell_to_input_weights->data.f : nullptr; in EvalFloat()
[all …]
|
| D | optional_tensor_test.cc | 34 bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
63 if (use_peephole) { in LSTMOpModel()
|
| D | lstm_test.cc | 38 bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
69 if (use_peephole) { in LSTMOpModel()
272 bool use_cifg, bool use_peephole, in HybridLSTMOpModel() argument
277 : LSTMOpModel(n_batch, n_input, n_cell, n_output, use_cifg, use_peephole, in HybridLSTMOpModel()
1634 bool use_cifg, bool use_peephole, in LayerNormLSTMOpModel() argument
1639 : LSTMOpModel(n_batch, n_input, n_cell, n_output, use_cifg, use_peephole, in LayerNormLSTMOpModel()
1648 bool use_cifg, bool use_peephole, in HybridLayerNormLSTMOpModel() argument
1655 use_peephole, use_projection_weights, in HybridLayerNormLSTMOpModel()
|
| D | unidirectional_sequence_lstm_test.cc | 36 bool use_peephole, bool use_projection_weights, in UnidirectionalLSTMOpModel() argument
68 if (use_peephole) { in UnidirectionalLSTMOpModel()
244 bool time_major, bool use_cifg, bool use_peephole, in HybridUnidirectionalLSTMOpModel() argument
250 use_cifg, use_peephole, use_projection_weights, use_projection_bias, in HybridUnidirectionalLSTMOpModel()
|
| D | bidirectional_sequence_lstm_test.cc | 39 bool use_peephole, bool use_projection_weights, in BidirectionalLSTMOpModel() argument
76 if (use_peephole) { in BidirectionalLSTMOpModel()
131 if (use_peephole) { in BidirectionalLSTMOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2260 bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
2291 if (use_peephole) { in LSTMOpModel()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 71 num_units, forget_bias=0, cell_clip=None, use_peephole=False,
|