Searched refs:n_unit (Results 1 – 3 of 3) sorted by relevance
45 const int n_unit = n_filter / n_rank; in EvalIntegerSvdfReference() local136 int32_t* output_ptr = scratch_output_tensor + i * n_unit; in EvalIntegerSvdfReference()138 for (int j = 0; j < n_unit; ++j) { in EvalIntegerSvdfReference()144 for (int i = 0; i < n_batch * n_unit; ++i) { in EvalIntegerSvdfReference()151 int32_t* output_temp_ptr = scratch_output_tensor + b * n_unit; in EvalIntegerSvdfReference()155 for (int i = 0; i < n_unit; ++i) { in EvalIntegerSvdfReference()165 for (int i = 0; i < n_batch * n_unit; ++i) { in EvalIntegerSvdfReference()
66 const int n_unit = n_filter / n_rank; in EvalIntegerSvdfHifimini() local202 int32_t* output_ptr = scratch_output_tensor + i * n_unit; in EvalIntegerSvdfHifimini()204 for (int j = 0; j < n_unit; ++j) { in EvalIntegerSvdfHifimini()210 for (int i = 0; i < n_batch * n_unit; ++i) { in EvalIntegerSvdfHifimini()217 int32_t* output_temp_ptr = scratch_output_tensor + b * n_unit; in EvalIntegerSvdfHifimini()221 for (int i = 0; i < n_unit; ++i) { in EvalIntegerSvdfHifimini()232 for (int i = 0; i < n_batch * n_unit; ++i) { in EvalIntegerSvdfHifimini()261 const int n_unit = n_filter / n_rank; in EvalIntegerSvdfHifi4() local303 tflite::micro::GetTensorData<int8_t>(output_tensor) + b * n_unit; in EvalIntegerSvdfHifi4()310 data.output_zero_point, n_unit), in EvalIntegerSvdfHifi4()
80 const int n_unit = n_filter / n_rank; in EvalIntegerSVDF() local133 n_batch * n_unit, n_rank); in EvalIntegerSVDF()136 tensor_utils::VectorBatchVectorAdd(bias_data, n_unit, n_batch, in EvalIntegerSVDF()142 for (int i = 0; i < n_batch * n_unit; ++i) { in EvalIntegerSVDF()