diff options
Diffstat (limited to 'src/core/GLES_COMPUTE/GCKernelLibrary.cpp')
-rw-r--r-- | src/core/GLES_COMPUTE/GCKernelLibrary.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/core/GLES_COMPUTE/GCKernelLibrary.cpp b/src/core/GLES_COMPUTE/GCKernelLibrary.cpp index 0b9cd3f4ee..d4ce3888fd 100644 --- a/src/core/GLES_COMPUTE/GCKernelLibrary.cpp +++ b/src/core/GLES_COMPUTE/GCKernelLibrary.cpp @@ -190,6 +190,7 @@ void GCKernel::update_shader_params() const std::map<std::string, std::string> GCKernelLibrary::_shader_program_map = { { "absdiff", "absdiff.cs" }, + { "tensorshift", "tensor_shift.cs" }, { "direct_convolution1x1", "direct_convolution1x1.cs" }, { "direct_convolution3x3", "direct_convolution3x3.cs" }, { "direct_convolution5x5", "direct_convolution5x5.cs" }, @@ -235,6 +236,10 @@ const std::map<std::string, std::string> GCKernelLibrary::_program_source_map = #include "./cs_shaders/absdiff.csembed" }, { + "tensor_shift.cs", +#include "./cs_shaders/tensor_shift.csembed" + }, + { "convolution_layer.cs", #include "./cs_shaders/convolution_layer.csembed" }, |