Home
last modified time | relevance | path

Searched refs:cell_clip (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Dlstm_ops.py53 cell_clip=None, argument
137 cell_clip=cell_clip if cell_clip is not None else -1,
153 cell_clip=None, argument
224 cell_clip=cell_clip if cell_clip is not None else -1,
351 cell_clip=None, argument
379 self._cell_clip = cell_clip if cell_clip is not None else -1
439 cell_clip=self._cell_clip,
603 cell_clip=None, argument
628 self._cell_clip = cell_clip if cell_clip is not None else -1
710 cell_clip=self._cell_clip,
Drnn_cell.py340 cell_clip=None, argument
371 self._cell_clip = cell_clip
519 cell_clip=None, argument
580 self._cell_clip = cell_clip
979 cell_clip=None, argument
1029 num_units, use_peepholes, share_time_frequency_weights, cell_clip,
2501 cell_clip=None, argument
2546 self._cell_clip = cell_clip
2847 cell_clip=None, argument
2884 self._cell_clip = cell_clip
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Dlstm_ops_test.py107 def blocks_match(sess, use_peephole, dtype=dtypes.float32, cell_clip=None): argument
146 cell_clip=cell_clip,
162 cell_clip=cell_clip,
170 cell_clip=cell_clip)
174 cell_clip=cell_clip,
406 cell_clip=None, argument
413 sess, use_peephole=use_peephole, dtype=dtype, cell_clip=cell_clip)
439 dtype, use_peephole=True, cell_clip=0.5, rtol=rtol, atol=atol)
474 cell_size, cell_clip=0, use_peephole=False, reuse=True,
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h198 float cell_clip; member
209 float cell_clip; member
220 float cell_clip; member
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops_gpu.cu.cc89 const float forget_bias, const float cell_clip, in lstm_gates() argument
95 T cell_clip_t = strict_cast<T>(cell_clip); in lstm_gates()
172 if (cell_clip > 0.0f) { in lstm_gates()
224 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA() argument
269 i.data(), f.data(), forget_bias, cell_clip, batch_size, cell_size)); in LSTMBlockCellFpropWithCUDA()
275 i.data(), f.data(), forget_bias, cell_clip, batch_size, cell_size)); in LSTMBlockCellFpropWithCUDA()
412 const float cell_clip, bool use_peephole, \
423 LSTMBlockCellFpropWithCUDA<T>(ctx, d, forget_bias, cell_clip, \
Dlstm_ops.cc47 const float forget_bias, const float cell_clip, bool use_peephole, in LSTMBlockCellFpropWithEigen() argument
102 if (cell_clip > 0.0f) { in LSTMBlockCellFpropWithEigen()
104 cs.binaryExpr(cs.constant(T(cell_clip)), Eigen::scalar_clip_op<T>()); in LSTMBlockCellFpropWithEigen()
185 const float cell_clip, bool use_peephole, \
197 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
387 const float cell_clip, bool use_peephole, \
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc389 params->cell_clip = lstm_params->cell_clip(); in ParseOpData()
410 params->cell_clip = seq_lstm_params->cell_clip(); in ParseOpData()
424 params->cell_clip = bidi_lstm_params->cell_clip(); in ParseOpData()
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Drnn_cells.py183 cell_clip=None, argument
229 cell_clip=cell_clip,
/external/tensorflow/tensorflow/lite/kernels/
Doptional_tensor_test.cc35 bool use_projection_bias, float cell_clip, float proj_clip, in LSTMOpModel() argument
108 cell_clip, proj_clip) in LSTMOpModel()
Dlstm_test.cc39 bool use_projection_bias, float cell_clip, float proj_clip, in LSTMOpModel() argument
126 cell_clip, proj_clip) in LSTMOpModel()
274 float cell_clip, float proj_clip, in HybridLSTMOpModel() argument
278 use_projection_weights, use_projection_bias, cell_clip, in HybridLSTMOpModel()
1636 float cell_clip, float proj_clip, in LayerNormLSTMOpModel() argument
1640 use_projection_weights, use_projection_bias, cell_clip, in LayerNormLSTMOpModel()
1650 bool use_projection_bias, float cell_clip, in HybridLayerNormLSTMOpModel() argument
1656 use_projection_bias, cell_clip, proj_clip, in HybridLayerNormLSTMOpModel()
Dunidirectional_sequence_lstm_test.cc37 bool use_projection_bias, float cell_clip, in UnidirectionalLSTMOpModel() argument
116 builder_, ActivationFunctionType_TANH, cell_clip, in UnidirectionalLSTMOpModel()
245 bool use_projection_weights, bool use_projection_bias, float cell_clip, in HybridUnidirectionalLSTMOpModel() argument
251 cell_clip, proj_clip, input_shapes, tensor_type) { in HybridUnidirectionalLSTMOpModel()
Dunidirectional_sequence_lstm.cc108 TF_LITE_ENSURE(context, params->cell_clip >= 0); in CheckInputTensorDimensions()
495 lstm_params.cell_clip = params->cell_clip; in Eval()
Dlstm_eval.cc268 if (params->cell_clip > 0.0) { in LstmStepWithAuxInput()
270 params->cell_clip, cell_state_ptr); in LstmStepWithAuxInput()
725 if (params->cell_clip > 0.0) { in LstmStepWithAuxInput()
727 params->cell_clip, cell_state_ptr); in LstmStepWithAuxInput()
Dbidirectional_sequence_lstm_test.cc41 bool use_aux_input, float cell_clip, float proj_clip, in BidirectionalLSTMOpModel() argument
205 builder_, ActivationFunctionType_TANH, cell_clip, in BidirectionalLSTMOpModel()
Dbidirectional_sequence_lstm.cc175 TF_LITE_ENSURE(context, params->cell_clip >= 0); in CheckLstmTensorDimensionsAndTypes()
881 TfLiteLSTMParams lstm_params = {params->activation, params->cell_clip, in Eval()
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py193 cell_clip=None, argument
261 self._cell_clip = cell_clip
/external/tensorflow/tensorflow/lite/schema/
Dschema.fbs456 cell_clip: float; // Optional, 0.0 means no clipping
467 cell_clip: float; // Optional, 0.0 means no clipping
477 cell_clip: float; // Optional, 0.0 means no clipping
Dschema_v0.fbs195 cell_clip: float; // Optional, 0.0 means no clipping
Dschema_v2.fbs216 cell_clip: float; // Optional, 0.0 means no clipping
Dschema_v1.fbs208 cell_clip: float; // Optional, 0.0 means no clipping
Dschema_generated.h3971 float cell_clip;
3976 cell_clip(0.0f),
3993 float cell_clip() const {
4021 void add_cell_clip(float cell_clip) {
4022 fbb_.AddElement<float>(LSTMOptions::VT_CELL_CLIP, cell_clip, 0.0f);
4045 float cell_clip = 0.0f,
4050 builder_.add_cell_clip(cell_clip);
4061 float cell_clip;
4066 cell_clip(0.0f),
4083 float cell_clip() const {
[all …]
Dschema_v3.fbs226 cell_clip: float; // Optional, 0.0 means no clipping
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt108 …argspec: "args=[\'self\', \'num_units\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_…
Dtensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt108 …argspec: "args=[\'self\', \'num_units\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt108 …argspec: "args=[\'self\', \'num_units\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_…

12