Home
last modified time | relevance | path

Searched refs:proj_clip (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
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
279 proj_clip, input_shapes, tensor_type) { in HybridLSTMOpModel()
1636 float cell_clip, float proj_clip, in LayerNormLSTMOpModel() argument
1641 proj_clip, input_shapes, weight_type, in LayerNormLSTMOpModel()
1651 float proj_clip, in HybridLayerNormLSTMOpModel() argument
1656 use_projection_bias, cell_clip, proj_clip, in HybridLayerNormLSTMOpModel()
1873 const float proj_clip = 0.0; in TEST_F() local
1879 /*use_projection_bias=*/false, ceil_clip, proj_clip, in TEST_F()
[all …]
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_eval.cc313 if (params->proj_clip > 0.0) { in LstmStepWithAuxInput()
315 params->proj_clip, output_ptr_batch); in LstmStepWithAuxInput()
343 if (params->proj_clip > 0.0) { in LstmStepWithAuxInput()
346 params->proj_clip, in LstmStepWithAuxInput()
791 if (params->proj_clip > 0.0) { in LstmStepWithAuxInput()
793 params->proj_clip, output_ptr_batch); in LstmStepWithAuxInput()
838 if (params->proj_clip > 0.0) { in LstmStepWithAuxInput()
842 params->proj_clip, in LstmStepWithAuxInput()
Dunidirectional_sequence_lstm_test.cc38 float proj_clip, in UnidirectionalLSTMOpModel() argument
117 proj_clip, time_major) in UnidirectionalLSTMOpModel()
246 float proj_clip, const std::vector<std::vector<int>>& input_shapes, in HybridUnidirectionalLSTMOpModel() argument
251 cell_clip, proj_clip, input_shapes, tensor_type) { in HybridUnidirectionalLSTMOpModel()
Dunidirectional_sequence_lstm.cc109 TF_LITE_ENSURE(context, params->proj_clip >= 0); in CheckInputTensorDimensions()
496 lstm_params.proj_clip = params->proj_clip; in Eval()
Dbidirectional_sequence_lstm_test.cc41 bool use_aux_input, float cell_clip, float proj_clip, in BidirectionalLSTMOpModel() argument
206 proj_clip, merge_outputs, time_major) in BidirectionalLSTMOpModel()
Dbidirectional_sequence_lstm.cc176 TF_LITE_ENSURE(context, params->proj_clip >= 0); in CheckLstmTensorDimensionsAndTypes()
882 params->proj_clip, kTfLiteLSTMFullKernel}; in Eval()
Dlstm.cc126 TF_LITE_ENSURE(context, params->proj_clip >= 0); in CheckInputTensorDimensions()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h199 float proj_clip; member
210 float proj_clip; member
221 float proj_clip; member
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc390 params->proj_clip = lstm_params->proj_clip(); in ParseOpData()
411 params->proj_clip = seq_lstm_params->proj_clip(); in ParseOpData()
425 params->proj_clip = bidi_lstm_params->proj_clip(); in ParseOpData()
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Drnn_cells.py186 proj_clip=None, argument
232 proj_clip=proj_clip,
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py196 proj_clip=None, argument
264 self._proj_clip = proj_clip
/external/tensorflow/tensorflow/lite/schema/
Dschema.fbs457 proj_clip: float; // Optional, 0.0 means no clipping
468 proj_clip: float; // Optional, 0.0 means no clipping
478 proj_clip: float; // Optional, 0.0 means no clipping
Dschema_v0.fbs196 proj_clip: float; // Optional, 0.0 means no clipping
Dschema_v2.fbs217 proj_clip: float; // Optional, 0.0 means no clipping
Dschema_v1.fbs209 proj_clip: float; // Optional, 0.0 means no clipping
Dschema_generated.h3972 float proj_clip;
3977 proj_clip(0.0f),
3996 float proj_clip() const {
4024 void add_proj_clip(float proj_clip) {
4025 fbb_.AddElement<float>(LSTMOptions::VT_PROJ_CLIP, proj_clip, 0.0f);
4046 float proj_clip = 0.0f,
4049 builder_.add_proj_clip(proj_clip);
4062 float proj_clip;
4067 proj_clip(0.0f),
4086 float proj_clip() const {
[all …]
Dschema_v3.fbs227 proj_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 …s\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\', \'num_unit_sh…
Dtensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt108 …s\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\', \'num_unit_sh…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt108 …s\', \'use_peepholes\', \'cell_clip\', \'initializer\', \'num_proj\', \'proj_clip\', \'num_unit_sh…
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Drnn_cell.py140 proj_clip=None, argument
190 self._proj_clip = proj_clip
2504 proj_clip=None, argument
2549 self._proj_clip = proj_clip
2850 proj_clip=None, argument
2886 self._proj_clip = proj_clip
/external/tensorflow/tensorflow/python/ops/
Drnn_cell_impl.py833 initializer=None, num_proj=None, proj_clip=None, argument
902 self._proj_clip = proj_clip
/external/tensorflow/tensorflow/contrib/lite/
Dnnapi_delegate.cc391 add_scalar_float32(builtin->proj_clip); in AddOpsAndParams()
/external/tensorflow/tensorflow/lite/
Dnnapi_delegate.cc361 add_scalar_float32(builtin->proj_clip); in AddOpsAndParams()

12