| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_test.cc | 39 bool use_projection_bias, float cell_clip, float proj_clip, in LSTMOpModel() argument
134 cell_clip, proj_clip) in LSTMOpModel()
1635 const float cell_clip = 0.0; in TEST_F() local
1642 /*use_projection_bias=*/false, cell_clip, proj_clip, in TEST_F()
1702 const float cell_clip = 0.0; in TEST_F() local
1709 /*use_projection_bias=*/false, cell_clip, proj_clip, in TEST_F()
1769 const float cell_clip = 0.0; in TEST_F() local
1776 /*use_projection_bias=*/false, cell_clip, proj_clip, in TEST_F()
1881 const float cell_clip = 0.0; in TEST_F() local
1888 /*use_projection_bias=*/false, cell_clip, proj_clip, in TEST_F()
[all …]
|
| D | optional_tensor_test.cc | 35 bool use_projection_bias, float cell_clip, float proj_clip, in LSTMOpModel() argument
108 cell_clip, proj_clip) in LSTMOpModel()
|
| D | unidirectional_sequence_lstm_test.cc | 37 bool use_projection_bias, float cell_clip, in UnidirectionalLSTMOpModel() argument
133 builder_, ActivationFunctionType_TANH, cell_clip, in UnidirectionalLSTMOpModel()
293 bool use_projection_weights, bool use_projection_bias, float cell_clip, in HybridUnidirectionalLSTMOpModel() argument
299 cell_clip, proj_clip, input_shapes, tensor_type) { in HybridUnidirectionalLSTMOpModel()
2470 bool use_projection_weights, bool use_projection_bias, float cell_clip, in LayerNormUnidirectionalLSTMOpModel() argument
2476 cell_clip, proj_clip, input_shapes, TensorType_FLOAT32, true) {} in LayerNormUnidirectionalLSTMOpModel()
|
| D | unidirectional_sequence_lstm.cc | 119 TF_LITE_ENSURE(context, params->cell_clip >= 0); in CheckInputTensorDimensions()
601 lstm_params.cell_clip = params->cell_clip; in Eval()
|
| D | lstm.cc | 66 const float cell_clip = params->cell_clip; in PopulateQuantizedLstmParams() local
78 if (cell_clip > 0.0) { in PopulateQuantizedLstmParams()
80 std::min(std::max(cell_clip / cell_params->scale->data[0], -32768.0f), in PopulateQuantizedLstmParams()
389 TF_LITE_ENSURE(context, params->cell_clip >= 0); in CheckInputTensorDimensions()
|
| D | lstm_eval.cc | 299 if (params->cell_clip > 0.0) { in LstmStepFloat()
301 params->cell_clip, cell_state_ptr); in LstmStepFloat()
730 if (params->cell_clip > 0.0) { in LstmStepHybrid()
732 params->cell_clip, cell_state_ptr); in LstmStepHybrid()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 225 float cell_clip; member
236 float cell_clip; member
247 float cell_clip; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | basic_lstm.mlir | 6 // CHECK: "tfl.basic_lstm"({{.*}}) {cell_clip = 1.000000e+00 : f32, fused_activation_function = "RE…
8 …%arg0, %arg1, %arg2, %arg3, %arg4) {fused_activation_function = "RELU", cell_clip = 1.0 : f32, pro…
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 88 const float forget_bias, const float cell_clip, in lstm_gates() argument
94 T cell_clip_t = strict_cast<T>(cell_clip); in lstm_gates()
176 if (cell_clip > 0.0f) { in lstm_gates()
229 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA() argument
274 co.data(), i.data(), f.data(), forget_bias, cell_clip, batch_size, in LSTMBlockCellFpropWithCUDA()
281 co.data(), i.data(), f.data(), forget_bias, cell_clip, batch_size, in LSTMBlockCellFpropWithCUDA()
414 const float cell_clip, bool use_peephole, \
426 ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, h_prev, w, \
|
| D | lstm_ops.cc | 47 const float forget_bias, const float cell_clip, bool use_peephole, in LSTMBlockCellFpropWithEigen() argument
106 if (cell_clip > 0.0f) { in LSTMBlockCellFpropWithEigen()
108 cs.binaryExpr(cs.constant(T(cell_clip)), Eigen::scalar_clip_op<T>()); in LSTMBlockCellFpropWithEigen()
192 const float cell_clip, bool use_peephole, \
204 *this, ctx, d, forget_bias, cell_clip, use_peephole, x, cs_prev, \
252 const float cell_clip, bool use_peephole, \
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 424 params->cell_clip = lstm_params->cell_clip(); in ParseOpData()
452 params->cell_clip = seq_lstm_params->cell_clip(); in ParseOpData()
466 params->cell_clip = bidi_lstm_params->cell_clip(); in ParseOpData()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LSTMBlockCell.pbtxt | 101 name: "cell_clip"
132 cs = clip(cs, cell_clip)
|
| D | api_def_BlockLSTM.pbtxt | 108 name: "cell_clip"
|
| D | api_def_BlockLSTMV2.pbtxt | 102 name: "cell_clip"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | basic_lstm.mlir | 83 // CHECK-NEXT: cell_clip: 1.0,
115 …%arg0, %arg1, %arg2, %arg3, %arg4) {fused_activation_function = "RELU", cell_clip = 1.0 : f32, pro…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BlockLSTMV2.pbtxt | 68 name: "cell_clip"
|
| D | LSTMBlockCell.pbtxt | 71 name: "cell_clip"
|
| D | BlockLSTM.pbtxt | 75 name: "cell_clip"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 54 cell_clip=0.0), # Disable to match V2 default.
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 195 cell_clip=None, argument
263 self._cell_clip = cell_clip
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-composite-functions-tf.mlir | 89 // CHECK: }) {cell_clip = 1.000000e+01 : f32, fused_activation_function = "TANH", kernel_…
149 // CHECK: }) {cell_clip = 1.000000e+01 : f32, fused_activation_function = "TANH", kernel_…
184 // CHECK: }) {cell_clip = 1.000000e+01 : f32, fused_activation_function = "TANH", kernel_…
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 220 if (params->cell_clip > 0.0) { in LstmStepWithAuxInput()
222 params->cell_clip, cell_state_ptr); in LstmStepWithAuxInput()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v0.fbs | 195 cell_clip: float; // Optional, 0.0 means no clipping
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt | 134 …argspec: "args=[\'self\', \'num_units\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_…
|
| D | tensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt | 134 …argspec: "args=[\'self\', \'num_units\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_…
|