| /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…
|
| D | lstm.mlir | 7 …15, %arg16, %arg17, %cst0, %cst1, %arg18, %arg19, %arg20, %arg21) ({}) {cell_clip = 0.000000e+00 :…
14 // CHECK: }) {cell_clip = 0.000000e+00 : f32, fused_activation_function = "NONE", kernel_type = "F…
23 …, %arg13, %arg14, %arg19, %arg20, %arg15, %arg16, %arg17, %arg18) ({}) {cell_clip = 1.000000e+01 :…
28 // CHECK: }) {cell_clip = 1.000000e+01 : f32, effective_hidden_scale_intermediate = tensor<0x!quant…
35 …arg15, %arg16, %arg17, %arg18, %arg19, %arg20, %arg21, %arg22, %arg23) {cell_clip = 1.000000e+01 :…
36 …arg15, %arg16, %arg17, %arg18, %arg19, %arg20, %arg21, %arg22, %arg23) {cell_clip = 1.000000e+01 :…
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 266 float cell_clip; member
280 float cell_clip; member
294 float cell_clip; member
|
| /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/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/lite/kernels/ |
| D | optional_tensor_test.cc | 33 bool use_projection_bias, float cell_clip, float proj_clip, in LSTMOpModel() argument
106 cell_clip, proj_clip) in LSTMOpModel()
|
| D | unidirectional_sequence_lstm.cc | 54 const float cell_clip = params->cell_clip; in PopulateQuantizedLstmParams8x8_16() local
69 if (cell_clip > 0.0) { in PopulateQuantizedLstmParams8x8_16()
71 std::max(cell_clip / cell_state_params->scale->data[0], -32768.0f), in PopulateQuantizedLstmParams8x8_16()
418 TF_LITE_ENSURE(context, params->cell_clip >= 0); in CheckInputTensorDimensions()
1310 lstm_params.cell_clip = params->cell_clip; in Eval()
|
| D | unidirectional_sequence_lstm_test.cc | 35 bool use_projection_bias, float cell_clip, in UnidirectionalLSTMOpModel() argument
130 builder_, ActivationFunctionType_TANH, cell_clip, in UnidirectionalLSTMOpModel()
290 bool use_projection_weights, bool use_projection_bias, float cell_clip, in HybridUnidirectionalLSTMOpModel() argument
296 cell_clip, proj_clip, input_shapes, tensor_type, false, in HybridUnidirectionalLSTMOpModel()
2472 bool use_projection_weights, bool use_projection_bias, float cell_clip, in LayerNormUnidirectionalLSTMOpModel() argument
2478 cell_clip, proj_clip, input_shapes, TensorType_FLOAT32, true) {} in LayerNormUnidirectionalLSTMOpModel()
|
| D | lstm.cc | 146 const float cell_clip = params->cell_clip; in PopulateQuantizedLstmParams8x8_16() local
160 if (cell_clip > 0.0) { in PopulateQuantizedLstmParams8x8_16()
162 std::max(cell_clip / cell_state_params->scale->data[0], -32768.0f), in PopulateQuantizedLstmParams8x8_16()
808 const float cell_clip = params->cell_clip; in PopulateQuantizedLstmParams8x8_8() local
820 if (cell_clip > 0.0 && cell_clip < 1.0) { in PopulateQuantizedLstmParams8x8_8()
822 std::max(cell_clip / cell_state_params->scale->data[0], -32768.0f), in PopulateQuantizedLstmParams8x8_8()
861 TF_LITE_ENSURE(context, params->cell_clip >= 0); in CheckInputTensorDimensions()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_grad_test.py | 54 cell_clip=0.0), # Disable to match V2 default.
97 cell_clip=0.0,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | basic_lstm.mlir | 84 // CHECK-NEXT: cell_clip: 1.0,
124 …%arg0, %arg1, %arg2, %arg3, %arg4) {fused_activation_function = "RELU", cell_clip = 1.0 : f32, pro…
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 252 Value* cell_gate, float cell_clip, in BuildCellStateUpdate() argument
286 if (cell_clip <= 0.0f) { in BuildCellStateUpdate()
297 attr.param = cell_clip; in BuildCellStateUpdate()
308 attr.param = -cell_clip; in BuildCellStateUpdate()
513 cell_gate, params->cell_clip, in ParseLSTMAttributes()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LSTMBlockCell.pbtxt | 71 name: "cell_clip"
|
| D | BlockLSTMV2.pbtxt | 68 name: "cell_clip"
|
| D | BlockLSTM.pbtxt | 75 name: "cell_clip"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | LSTMBlockCell.pbtxt | 71 name: "cell_clip"
|
| D | BlockLSTMV2.pbtxt | 68 name: "cell_clip"
|
| D | BlockLSTM.pbtxt | 75 name: "cell_clip"
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 238 params->cell_clip); in LstmStepCalibration()
594 lstm_params.cell_clip = params->cell_clip; in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 200 cell_clip=None, argument
268 self._cell_clip = cell_clip
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 567 params->cell_clip = lstm_params->cell_clip(); in ParseOpDataTfLite()
600 params->cell_clip = seq_lstm_params->cell_clip(); in ParseOpDataTfLite()
617 params->cell_clip = bidi_lstm_params->cell_clip(); in ParseOpDataTfLite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-quantize-post-training.mlir | 21 …%cst_2, %cst_2, %cst_2, %cst_2) {cell_clip = 1.000000e+01 : f32, fused_activation_function = "TANH…
76 %none, %none, %none, %none) {cell_clip = 5.000000e+01 : f32,
151 cell_clip = 5.000000e+01 : f32,
236 %19, %20, %21, %22) {cell_clip = 5.000000e+01 : f32,
328 cell_clip = 5.000000e+01 : f32,
|