diff options
Diffstat (limited to 'inference-engine/thirdparty/clDNN/kernel_selector/core/common/common_kernel_base.h')
| -rw-r--r-- | inference-engine/thirdparty/clDNN/kernel_selector/core/common/common_kernel_base.h | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/common/common_kernel_base.h b/inference-engine/thirdparty/clDNN/kernel_selector/core/common/common_kernel_base.h index 324cd9d67..917b4e5ac 100644 --- a/inference-engine/thirdparty/clDNN/kernel_selector/core/common/common_kernel_base.h +++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/common/common_kernel_base.h @@ -18,8 +18,6 @@ #include "kernel_base.h" #include "jitter.h" -#include <sstream> -#include <assert.h> namespace kernel_selector { @@ -51,7 +49,7 @@ namespace kernel_selector std::string CreateJit(const std::string& template_name, const JitConstants& constants, const std::string& kernel_name) const; std::string GetEntryPoint(const std::string& templateName, const std::string& layerID, const optional_params& options) const; Arguments GetArgsDesc(uint32_t num_of_input, bool use_weights, bool use_bias, bool use_quantization = false, bool use_calibration = 0) const; - std::shared_ptr<KernelString> GetKernelString(const std::string& kernel_name, const std::string& jit, const std::string& entry_point, const std::string& exe_mode = ROUND_ROBIN) const; - void FillCLKernelData(clKernelData& kernel, const CommonDispatchData& runInfo, const std::string& kernel_map_name, const std::string& jit, const std::string& entry_point, const std::string& exe_mode = ROUND_ROBIN, + std::shared_ptr<KernelString> GetKernelString(const std::string& kernel_name, const std::string& jit, const std::string& entry_point, const EngineInfo& engine_info, const std::string& exe_mode = ROUND_ROBIN) const; + void FillCLKernelData(clKernelData& kernel, const CommonDispatchData& runInfo, const EngineInfo& engine_info, const std::string& kernel_map_name, const std::string& jit, const std::string& entry_point, const std::string& exe_mode = ROUND_ROBIN, bool weights = false, bool bias = false, int number_of_inputs = 1, bool quantization = false, bool calibration = false) const; }; }
\ No newline at end of file |