| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 84 template <typename T, bool use_peephole, GateLayout gate_layout>
149 if (use_peephole) { in lstm_gates()
166 if (use_peephole) { in lstm_gates()
185 if (use_peephole) { in lstm_gates()
229 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA() argument
269 if (use_peephole) { in LSTMBlockCellFpropWithCUDA()
310 const int batch_size, const int cell_size, const bool use_peephole) { in lstm_gates_bprop() argument
335 if (use_peephole) { in lstm_gates_bprop()
358 if (use_peephole) { in lstm_gates_bprop()
380 const bool use_peephole) { in LSTMBlockCellBpropWithCUDA() argument
[all …]
|
| D | lstm_ops.h | 170 bool use_peephole, typename TTypes<T>::ConstMatrix x,
194 OpKernelContext* ctx, const Device& d, bool use_peephole,
219 OpKernelContext* ctx, const Device& d, bool use_peephole, in operator()
248 if (use_peephole) { in operator()
268 if (use_peephole) { in operator()
297 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()
90 if (use_peephole) { in LSTMBlockCellFpropWithEigen()
115 if (use_peephole) { in LSTMBlockCellFpropWithEigen()
132 bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellBpropWithEigen() argument
155 if (use_peephole) { in LSTMBlockCellBpropWithEigen()
177 if (use_peephole) { in LSTMBlockCellBpropWithEigen()
192 const float cell_clip, bool use_peephole, \
204 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
210 OpKernelContext* ctx, const CPUDevice& d, bool use_peephole, \
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | operator_property.cc | 43 op_variant.use_peephole = op->inputs[cell_to_output_weight_index] != -1; in GetOperatorVariant()
270 op_variant.use_peephole) { in GetOperatorProperty()
341 !op_variant.use_peephole) { in GetOperatorProperty()
407 op_variant.use_peephole) { in GetOperatorProperty()
478 !op_variant.use_peephole) { in GetOperatorProperty()
543 op_variant.use_peephole) { in GetOperatorProperty()
613 !op_variant.use_peephole) { in GetOperatorProperty()
677 op_variant.use_peephole) { in GetOperatorProperty()
746 !op_variant.use_peephole) { in GetOperatorProperty()
|
| D | operator_property.h | 135 bool use_peephole = false; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 98 use_peephole=False)
117 use_peephole=False)
135 use_peephole=False)
|
| D | rnn_grad.py | 49 use_peephole=op.get_attr("use_peephole"))
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LSTMBlockCell.pbtxt | 107 name: "use_peephole"
124 if not use_peephole:
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | optional_tensor_test.cc | 32 bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
61 if (use_peephole) { in LSTMOpModel()
|
| D | unidirectional_sequence_lstm_test.cc | 34 bool use_peephole, bool use_projection_weights, in UnidirectionalLSTMOpModel() argument
68 if (use_peephole) { in UnidirectionalLSTMOpModel()
289 bool time_major, bool use_cifg, bool use_peephole, in HybridUnidirectionalLSTMOpModel() argument
295 use_cifg, use_peephole, use_projection_weights, use_projection_bias, in HybridUnidirectionalLSTMOpModel()
2471 bool time_major, bool use_cifg, bool use_peephole, in LayerNormUnidirectionalLSTMOpModel() argument
2477 use_cifg, use_peephole, use_projection_weights, use_projection_bias, in LayerNormUnidirectionalLSTMOpModel()
2746 bool time_major, bool use_cifg, bool use_peephole, in UnidirectionalSequenceLSTMIntegerOpModel() argument
2800 if (use_peephole) { in UnidirectionalSequenceLSTMIntegerOpModel()
|
| D | lstm_test.cc | 40 bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
68 if (use_peephole) { in LSTMOpModel()
1315 bool use_cifg, bool use_peephole, in LSTMIntegerOpModel() argument
1368 if (use_peephole) { in LSTMIntegerOpModel()
2131 bool use_peephole, bool use_projection_weights, bool use_projection_bias, in HybridSparseLSTMOpModel() argument
2181 if (use_peephole) { in HybridSparseLSTMOpModel()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BlockLSTMV2.pbtxt | 75 name: "use_peephole"
|
| D | LSTMBlockCell.pbtxt | 78 name: "use_peephole"
|
| D | BlockLSTM.pbtxt | 82 name: "use_peephole"
|
| D | LSTMBlockCellGrad.pbtxt | 88 name: "use_peephole"
|
| D | BlockLSTMGrad.pbtxt | 108 name: "use_peephole"
|
| D | BlockLSTMGradV2.pbtxt | 108 name: "use_peephole"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | LSTMBlockCell.pbtxt | 78 name: "use_peephole"
|
| D | BlockLSTMV2.pbtxt | 75 name: "use_peephole"
|
| D | BlockLSTM.pbtxt | 82 name: "use_peephole"
|
| D | LSTMBlockCellGrad.pbtxt | 88 name: "use_peephole"
|
| D | BlockLSTMGradV2.pbtxt | 108 name: "use_peephole"
|
| D | BlockLSTMGrad.pbtxt | 108 name: "use_peephole"
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 73 const bool use_peephole = (cell_to_output_weights_ptr != nullptr); in LstmStepWithAuxInput() local
198 if (use_peephole) { in LstmStepWithAuxInput()
219 if (use_peephole) { in LstmStepWithAuxInput()
269 if (use_peephole) { in LstmStepWithAuxInput()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 36 bool use_peephole, bool use_projection_weights, in LSTMOpModel() argument
88 if (use_peephole) { in LSTMOpModel()
|