Searched defs:gate_bias (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru.cc | 31 const TfLiteTensor* gate_weight, const TfLiteTensor* gate_bias, in GruImpl()
154 const TfLiteTensor* gate_bias; in Prepare() local
234 const TfLiteTensor* gate_bias; in Eval() local
|
| D | gru_cell.cc | 35 const RuntimeShape& gate_bias_shape, const float* gate_bias, in GruCell()
|
| /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()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 155 const float* layer_norm_coefficients, const float* gate_bias, in CalculateLstmGateFloat()
319 const float* layer_norm_coefficients, const float* gate_bias, in CalculateLstmGateHybrid()
669 const int32_t layer_norm_gate_scale_b, const int32_t* gate_bias, in CalculateLstmGateInteger8x8_8()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.h | 165 const int32_t* gate_bias, int32_t n_batch, in MatrixBatchVectorMultiply()
|
| D | portable_tensor_utils.cc | 377 const int32_t* gate_bias, int32_t n_batch, int32_t n_hidden, in PortableMatrixBatchVectorMultiply()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | sse_tensor_utils.h | 135 const int32_t* gate_bias, int32_t n_batch, in MatrixBatchVectorMultiply()
|
| D | neon_tensor_utils.h | 126 const int32_t* gate_bias, int32_t n_batch, in MatrixBatchVectorMultiply()
|