diff options
Diffstat (limited to 'inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/batch_norm')
-rw-r--r-- | inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/batch_norm/batch_norm_kernel_base.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/batch_norm/batch_norm_kernel_base.cpp b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/batch_norm/batch_norm_kernel_base.cpp index 811d4412f..ebf881f50 100644 --- a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/batch_norm/batch_norm_kernel_base.cpp +++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/batch_norm/batch_norm_kernel_base.cpp @@ -80,7 +80,7 @@ namespace kernel_selector auto& kernel = kd.kernels[0]; int inputs_num = 1 + orgParams.batchNormParams.with_inv_var; - FillCLKernelData(kernel, runInfo, kernelName, jit, entry_point, "", false, false, inputs_num); + FillCLKernelData(kernel, runInfo, params.engineInfo, kernelName, jit, entry_point, "", false, false, inputs_num); kd.estimatedTime = estimatedTime; |