Searched refs:VectorBatchVectorAssign (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 87 tensor_utils::VectorBatchVectorAssign(input_gate_bias_ptr, n_cell, in LstmStepWithAuxInput()
90 tensor_utils::VectorBatchVectorAssign(forget_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
92 tensor_utils::VectorBatchVectorAssign(cell_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
94 tensor_utils::VectorBatchVectorAssign(output_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | kernel_utils.cc | 50 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
185 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
|
| D | tensor_utils_common.h | 414 void VectorBatchVectorAssign(const T* vector, int v_size, int n_batch, in VectorBatchVectorAssign() function
|
| D | tensor_utils_test.cc | 1690 VectorBatchVectorAssign(input, kVectorSize, kBatchSize, output.data()); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 60 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
142 tensor_utils::VectorBatchVectorAssign(projection_bias, n_output, n_batch, in CalculateLstmOutputCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 168 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateFloat()
279 tensor_utils::VectorBatchVectorAssign(projection_bias, n_output, n_batch, in CalculateLstmOutputFloat()
343 tensor_utils::VectorBatchVectorAssign(gate_bias, n_cell, n_batch, gate); in CalculateLstmGateHybrid()
451 tensor_utils::VectorBatchVectorAssign(projection_bias, n_output, n_batch, in CalculateLstmOutputHybrid()
|
| D | fully_connected.cc | 414 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), num_units, in EvalPie()
466 tensor_utils::VectorBatchVectorAssign(GetTensorData<float>(bias), in EvalHybridImpl()
|