Spaces:
Running
Running
File size: 592 Bytes
81efcf0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
// Copyright (C) 2018-2022 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#pragma OPENCL EXTENSION cl_khr_fp16 : enable
__kernel void cvtf32f16(const __global float* restrict inImage,
__global half* restrict outImage,
float scale,
float bais)
{
int idx = get_global_id(0)
+ get_global_id(1) * get_global_size(0)
+ get_global_id(2) * get_global_size(0) * get_global_size(1);
outImage[idx] = convert_half(inImage[idx]*scale+bais);
}
|