C# (CSharp) Pix2Pix GpuHelper.InvokeBatchNormKernel Examples

Example #1

File: Math.cs Project: joshnewnham/Pix2Pix

        public static Tensor BatchNorm(Tensor input, Tensor scale, Tensor offset)
        {
            var channels = scale.Shape[0];
            var kernel   = channels == 512 ? "BatchNorm512" : "BatchNorm64";

            return(GpuHelper.InvokeBatchNormKernel(kernel, input, scale, offset));
        }

Example #2

File: Math.cs Project: L4Z3RC47/Pix2Pix

        public static Tensor BatchNorm(Tensor input, Tensor scale, Tensor offset)
        {
            var kernel = "BatchNorm" + scale.Shape[0];

            return(GpuHelper.InvokeBatchNormKernel(kernel, input, scale, offset));
        }

Example #3

 public static Tensor BatchNorm(Tensor input, Tensor scale, Tensor offset)
 {
     return(GpuHelper.InvokeBatchNormKernel(input, scale, offset));
 }