Searched refs:fused_kernel (Results 1 – 6 of 6) sorted by relevance
65 params.fused_kernel.kernel.multiplicative_offset = multiplicative_offset; in gemm_q8_strided()66 params.fused_kernel.kernel.rounding_offset = (1 << (shift - 1)); in gemm_q8_strided()67 params.fused_kernel.kernel.shift = -shift; in gemm_q8_strided()68 params.fused_kernel.kernel.count = k; in gemm_q8_strided()69 params.fused_kernel.output_stream.stride = result_stride; in gemm_q8_strided()109 params.fused_kernel.kernel.multiplicative_offset = multiplicative_offset; in gemv_q8()110 params.fused_kernel.kernel.rounding_offset = (1 << (shift - 1)); in gemv_q8()111 params.fused_kernel.kernel.shift = -shift; in gemv_q8()112 params.fused_kernel.kernel.count = k; in gemv_q8()113 params.fused_kernel.output_stream.stride = n; in gemv_q8()[all …]
99 OutputStreamFF::Debug(params.fused_kernel.output_stream); in ExecuteDispatch3D()100 OutputStreamLF::Debug(params.fused_kernel.output_stream); in ExecuteDispatch3D()102 KernelFF::Debug(params.fused_kernel); in ExecuteDispatch3D()103 KernelFL::Debug(params.fused_kernel); in ExecuteDispatch3D()104 KernelLF::Debug(params.fused_kernel); in ExecuteDispatch3D()105 KernelLL::Debug(params.fused_kernel); in ExecuteDispatch3D()155 OutputStreamFF::UnpackedStride(params.fused_kernel.output_stream); in ExecuteDispatch3D()157 OutputStreamFF::UnpackedAdvance(params.fused_kernel.output_stream); in ExecuteDispatch3D()170 params.fused_kernel, in ExecuteDispatch3D()179 params.fused_kernel, in ExecuteDispatch3D()[all …]
264 params->fused_kernel.kernel.count = k; in setup_row_row()265 params->fused_kernel.kernel.multiplicative_offset = MUL_OFFSET; in setup_row_row()266 params->fused_kernel.kernel.rounding_offset = (1 << (SHIFT - 1)); in setup_row_row()267 params->fused_kernel.kernel.shift = -SHIFT; in setup_row_row()268 params->fused_kernel.output_stream.stride = n; in setup_row_row()281 params->fused_kernel.kernel.count = k; in setup_row_col()282 params->fused_kernel.kernel.multiplicative_offset = MUL_OFFSET; in setup_row_col()283 params->fused_kernel.kernel.rounding_offset = (1 << (SHIFT - 1)); in setup_row_col()284 params->fused_kernel.kernel.shift = -SHIFT; in setup_row_col()285 params->fused_kernel.output_stream.stride = n; in setup_row_col()[all …]
73 FusedKernel fused_kernel; member
49 Offset(params.fused_kernel.output_stream, params.result, m_start, in PrepareGemmTask()
148 params.fused_kernel.kernel.count = k; in QuantizedGemmImpl()149 params.fused_kernel.output_stream.stride = ldc * sizeof(int32_t); in QuantizedGemmImpl()