| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| lstm_ops.py | 50 cell_clip=None,
73 cs = clip(cs, cell_clip)
96 cell_clip: An optional `float`. Defaults to `-1` (no clipping).
134 cell_clip=cell_clip if cell_clip is not None else -1,
150 cell_clip=None,
166 cell_clip: An optional `float`. Defaults to `-1` (no clipping).
220 cell_clip=cell_clip if cell_clip is not None else -1
[all...] |
| rnn_cell.py | 333 cell_clip=None,
345 cell_clip: (optional) A float value, if provided the cell state is clipped
364 self._cell_clip = cell_clip
512 cell_clip=None,
532 cell_clip: (optional) A float value, default None, if provided the cell
573 self._cell_clip = cell_clip
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| lstm_ops_gpu.cu.cc | 46 const T cell_clip, const int batch_size,
126 if (cell_clip > 0.0) {
127 cs_local = clip_op(cs_local, cell_clip);
178 const T cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x,
220 i.data(), f.data(), forget_bias, cell_clip, batch_size, cell_size);
225 i.data(), f.data(), forget_bias, cell_clip, batch_size, cell_size);
361 const T cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, \
371 LSTMBlockCellFpropWithCUDA(ctx, d, forget_bias, cell_clip, use_peephole, \
|
| lstm_ops.cc | 47 const T forget_bias, const T cell_clip, bool use_peephole,
101 if (cell_clip > 0.0f) {
103 cs.binaryExpr(cs.constant(cell_clip), Eigen::scalar_clip_op<T>());
184 const T cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, \
195 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
236 OP_REQUIRES_OK(ctx, ctx->GetAttr("cell_clip", &cell_clip_));
384 const T cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, \
[all...] |
| lstm_ops.h | 159 const T cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x,
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| builtin_op_data.h | 149 float cell_clip; member in struct:__anon39206
|
| nnapi_delegate.cc | 232 add_scalar_float32(builtin->cell_clip);
|
| model.cc | 464 params->cell_clip = lstm_params->cell_clip();
|
| /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
| rnn_cells.py | 183 cell_clip=None,
198 cell_clip: (optional) A float value, if provided the cell state is clipped
229 cell_clip=cell_clip,
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| optional_tensor_test.cc | 35 bool use_projection_bias, float cell_clip, float proj_clip,
107 cell_clip, proj_clip)
257 /*cell_clip=*/0.0, /*proj_clip=*/0.0,
|
| lstm.cc | 82 TF_LITE_ENSURE(context, params->cell_clip >= 0);
460 if (params->cell_clip > 0.0) {
462 params->cell_clip, cell_state->data.f);
|
| unidirectional_sequence_lstm.cc | 82 TF_LITE_ENSURE(context, params->cell_clip >= 0);
469 if (params->cell_clip > 0.0) {
471 params->cell_clip, cell_state->data.f);
|
| lstm_test.cc | 37 bool use_projection_bias, float cell_clip, float proj_clip,
109 cell_clip, proj_clip)
255 /*cell_clip=*/0.0, /*proj_clip=*/0.0,
365 /*cell_clip=*/0.0, /*proj_clip=*/0.0,
469 /*cell_clip=*/0.0, /*proj_clip=*/0.0,
[all...] |
| unidirectional_sequence_lstm_test.cc | 38 bool use_projection_bias, float cell_clip,
113 cell_clip, proj_clip)
261 /*use_projection_bias=*/false, /*cell_clip=*/0.0, /*proj_clip=*/0.0,
369 /*use_projection_bias=*/false, /*cell_clip=*/0.0, /*proj_clip=*/0.0,
471 /*cell_clip=*/0.0, /*proj_clip=*/0.0,
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| lstm_ops_test.py | 90 cell_clip=0,
98 cell_clip=0)
101 cell_size, cell_clip=0, use_peephole=use_peephole, reuse=True,
386 cell_size, cell_clip=0, use_peephole=False, reuse=True,
|
| core_rnn_test.py | 343 cell_clip=0.0,
[all...] |
| /frameworks/ml/nn/common/operations/ |
| LSTMTest.cpp | 79 bool use_projection_bias, float cell_clip, float proj_clip,
87 cell_clip_(cell_clip), proj_clip_(proj_clip) {
284 /*cell_clip=*/0.0, /*proj_clip=*/0.0,
395 /*cell_clip=*/0.0, /*proj_clip=*/0.0,
499 /*cell_clip=*/0.0, /*proj_clip=*/0.0,
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| rnn_cell_impl.py | 624 use_peepholes=False, cell_clip=None,
634 cell_clip: (optional) A float value, if provided the cell state is clipped
680 self._cell_clip = cell_clip
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/schema/ |
| schema_generated.h | 2466 float cell_clip; member in struct:tflite::LSTMOptionsT
2485 float cell_clip() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | 74 num_units, forget_bias=0, cell_clip=None, use_peephole=False,
[all...] |
