| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru.cc | 31 const TfLiteTensor* gate_weight, const TfLiteTensor* gate_bias, in GruImpl() argument
49 const RuntimeShape gate_bias_shape = GetTensorShape(gate_bias); in GruImpl()
50 const float* gate_bias_data = GetTensorData<float>(gate_bias); in GruImpl()
154 const TfLiteTensor* gate_bias; in Prepare() local
156 GetInputSafe(context, node, kGateBias, &gate_bias)); in Prepare()
157 TF_LITE_ENSURE_EQ(context, gate_bias->dims->size, 1); in Prepare()
158 TF_LITE_ENSURE_EQ(context, gate_bias->dims->data[0], 2 * n_output); in Prepare()
234 const TfLiteTensor* gate_bias; in Eval() local
236 GetInputSafe(context, node, kGateBias, &gate_bias)); in Eval()
256 GruImpl(input, input_state, gate_weight, gate_bias, candidate_weight, in Eval()
|
| D | gru_cell.cc | 35 const RuntimeShape& gate_bias_shape, const float* gate_bias, in GruCell() argument
64 gate_weight, gate_bias_shape, gate_bias, activation_shape, in GruCell()
|
| D | gru_cell.h | 30 const RuntimeShape& gate_bias_shape, const float* gate_bias,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.h | 165 const int32_t* gate_bias, int32_t n_batch, in MatrixBatchVectorMultiply() argument
170 proj_effective_scale_b, gate_bias, n_batch, in MatrixBatchVectorMultiply()
|
| D | portable_tensor_utils_impl.h | 132 const int32_t* gate_bias, int32_t n_batch, int32_t n_hidden,
|
| D | portable_tensor_utils.cc | 377 const int32_t* gate_bias, int32_t n_batch, int32_t n_hidden, in PortableMatrixBatchVectorMultiply() argument
383 int64_t acc = gate_bias[row]; in PortableMatrixBatchVectorMultiply()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | neon_tensor_utils.h | 126 const int32_t* gate_bias, int32_t n_batch, in MatrixBatchVectorMultiply() argument
131 proj_effective_scale_b, gate_bias, n_batch, in MatrixBatchVectorMultiply()
|
| D | sse_tensor_utils.h | 135 const int32_t* gate_bias, int32_t n_batch, in MatrixBatchVectorMultiply() argument
140 proj_effective_scale_b, gate_bias, n_batch, in MatrixBatchVectorMultiply()
|
| /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
60 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
91 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 155 const float* layer_norm_coefficients, const float* gate_bias, in CalculateLstmGateFloat() argument
168 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
196 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
319 const float* layer_norm_coefficients, const float* gate_bias, in CalculateLstmGateHybrid() argument
343 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateHybrid()
407 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateHybrid()
669 const int32_t layer_norm_gate_scale_b, const int32_t* gate_bias, in CalculateLstmGateInteger8x8_8() argument
696 layer_norm_gate_scale_b, gate_bias, n_batch, n_cell, gate); in CalculateLstmGateInteger8x8_8()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_common.h | 172 const int32_t* gate_bias, int32_t n_batch,
|