Searched refs:VectorBatchVectorAdd (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | svdf.h | 57 tensor_utils::VectorBatchVectorAdd(bias_ptr, num_units, batch_size, in ApplyTimeWeightsBiasAndActivation()
136 tensor_utils::VectorBatchVectorAdd(bias_data, n_unit, n_batch, in EvalIntegerSVDF()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 211 tensor_utils::VectorBatchVectorAdd(input_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
232 tensor_utils::VectorBatchVectorAdd(forget_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
249 tensor_utils::VectorBatchVectorAdd(cell_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
282 tensor_utils::VectorBatchVectorAdd(output_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_common.h | 402 void VectorBatchVectorAdd(const T* vector, int v_size, int n_batch, in VectorBatchVectorAdd() function
|
| D | tensor_utils_test.cc | 1680 VectorBatchVectorAdd(input, kVectorSize, kBatchSize, output.data()); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 91 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 196 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
407 tensor_utils::VectorBatchVectorAdd(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateHybrid()
|