diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/core/CL/kernels/CLSoftmaxLayerKernel.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/core/CL/kernels/CLSoftmaxLayerKernel.cpp b/src/core/CL/kernels/CLSoftmaxLayerKernel.cpp index 0e81fc7aa4..da3b9423d5 100644 --- a/src/core/CL/kernels/CLSoftmaxLayerKernel.cpp +++ b/src/core/CL/kernels/CLSoftmaxLayerKernel.cpp @@ -68,6 +68,10 @@ void CLLogits1DMaxKernel::configure(const ICLTensor *input, ICLTensor *output) { build_opts.emplace(("-DFIXED_POINT_POSITION=" + support::cpp11::to_string(input->info()->fixed_point_position()))); } + else if(input->info()->data_type() == DataType::F16) + { + build_opts.emplace("-DUSE_F16"); + } // Tell the kernel that the width is not a multiple of 16 if((input->info()->dimension(0) % max_cl_vector_width) != 0) @@ -130,6 +134,10 @@ void CLLogits1DShiftExpSumKernel::configure(const ICLTensor *input, const ICLTen { build_opts.emplace(("-DFIXED_POINT_POSITION=" + support::cpp11::to_string(input->info()->fixed_point_position()))); } + else if(input->info()->data_type() == DataType::F16) + { + build_opts.emplace("-DUSE_F16"); + } // Tell the kernel that the width is not a multiple of 16 if((input->info()->dimension(0) % max_cl_vector_width) != 0) |