Searched refs:layer_norm_coefficients (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 155 const float* layer_norm_coefficients, const float* gate_bias, in CalculateLstmGateFloat() argument
161 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat()
194 tensor_utils::VectorBatchVectorCwiseProduct(layer_norm_coefficients, n_cell, in CalculateLstmGateFloat()
319 const float* layer_norm_coefficients, const float* gate_bias, in CalculateLstmGateHybrid() argument
336 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateHybrid()
405 tensor_utils::VectorBatchVectorCwiseProduct(layer_norm_coefficients, n_cell, in CalculateLstmGateHybrid()
501 const int16_t* layer_norm_coefficients, const int32_t* layer_norm_bias, in CalculateLstmGateInteger8x8_16() argument
515 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateInteger8x8_16()
541 gate, layer_norm_coefficients, layer_norm_bias, in CalculateLstmGateInteger8x8_16()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 45 const float* layer_norm_coefficients, const float* gate_bias, in CalculateLstmGateFloat() argument
53 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat()
89 tensor_utils::VectorBatchVectorCwiseProduct(layer_norm_coefficients, n_cell, in CalculateLstmGateFloat()
|