Home
last modified time | relevance | path

Searched defs:use_layer_norm (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc40 bool use_layer_norm; member
85 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16() local
412 bool use_layer_norm, bool is_integer) { in CheckInputTensorDimensions()
853 bool use_layer_norm = false; in Prepare() local
1210 const bool use_layer_norm = op_data->use_layer_norm; in Eval() local
Dlstm.cc52 bool use_layer_norm; member
176 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16() local
855 bool use_layer_norm, bool is_integer) { in CheckInputTensorDimensions()
1313 const bool use_layer_norm = op_data->use_layer_norm; in Prepare() local
Dlstm_eval.cc161 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat() local
336 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateHybrid() local
515 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateInteger8x8_16() local
Dunidirectional_sequence_lstm_test.cc2748 bool use_layer_norm, bool use_8x8_8_implementation, in UnidirectionalSequenceLSTMIntegerOpModel()
Dlstm_test.cc1317 bool use_layer_norm, bool use_8x8_8_implementation, in LSTMIntegerOpModel()
/external/tensorflow/tensorflow/lite/tools/optimize/
Doperator_property.h133 bool use_layer_norm = false; member
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc74 const bool use_layer_norm = (forget_layer_norm_coefficients_ptr != nullptr); in LstmStepWithAuxInput() local
523 bool use_layer_norm; member
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc53 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat() local
455 bool use_layer_norm; member
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3163 const bool use_layer_norm = input_shapes.size() > 20; in LSTMOpModel() local