Searched refs:right_stream (Results 1 – 6 of 6) sorted by relevance
60 params.right_stream.count = k; in gemm_q8_strided()61 params.right_stream.stride = k; in gemm_q8_strided()62 params.right_stream.multiplicative_sum_offset = lhs_offset; in gemm_q8_strided()63 params.right_stream.additive_sum_offset = 0; in gemm_q8_strided()104 params.right_stream.count = k; in gemv_q8()105 params.right_stream.stride = k; in gemv_q8()106 params.right_stream.multiplicative_sum_offset = lhs_offset; in gemv_q8()107 params.right_stream.additive_sum_offset = 0; in gemv_q8()152 params.right_stream.count = k; in gemm_i32_strided()153 params.right_stream.stride = k; in gemm_i32_strided()[all …]
40 params.right_stream, kernel_n, kernel_k); in EstimateScratchSize()96 RightStreamF::Debug(params.right_stream); in ExecuteDispatch3D()97 RightStreamL::Debug(params.right_stream); in ExecuteDispatch3D()122 RightStreamF::PackedStride(params.right_stream); in ExecuteDispatch3D()128 RightStreamF::UnpackedStride(params.right_stream); in ExecuteDispatch3D()132 params.right_stream, in ExecuteDispatch3D()140 params.right_stream, in ExecuteDispatch3D()229 params.right_stream, kernel_n, kernel_k); in EstimateScratchSize()284 RightStreamF::Debug(params.right_stream); in ExecuteDispatch3D()285 RightStreamL::Debug(params.right_stream); in ExecuteDispatch3D()[all …]
250 params->right_stream.multiplicative_sum_offset = LHS_OFFSET; in setup_params()251 params->right_stream.additive_sum_offset = 0; in setup_params()262 params->right_stream.count = k; in setup_row_row()263 params->right_stream.stride = k; in setup_row_row()279 params->right_stream.count = k; in setup_row_col()280 params->right_stream.stride = n; in setup_row_col()295 params->right_stream.count = k; in setup_row_row_f()296 params->right_stream.stride = k; in setup_row_row_f()309 params->right_stream.count = k; in setup_row_col_f()310 params->right_stream.stride = n; in setup_row_col_f()[all …]
72 RightStream right_stream; member
45 params.right_stream, params.rhs, n_start, 0); in PrepareGemmTask()
143 params.right_stream.count = k; in QuantizedGemmImpl()144 params.right_stream.stride = ldb; in QuantizedGemmImpl()145 params.right_stream.multiplicative_sum_offset = offset_a; in QuantizedGemmImpl()146 params.right_stream.additive_sum_offset = 0; in QuantizedGemmImpl()