tensorflow/tfjs tfjs-backend-webgpu-v0.0.1-alpha.7

on GitHub

tfjs-backend-webgpu (0.0.1-alpha.5 ==> 0.0.1-alpha.7)

Features

• expose CPU_HANDOFF_SIZE_THRESHOLD as a configurable flag (#5106).
• [webgpu] Add support for gatherND (#5057). Thanks, @haoyunfeix.
• [webgpu] Add support for sqrt (#5058). Thanks, @haoyunfeix.
• [webgpu] Add resizeNearestNeighbor support (#5059). Thanks, @haoyunfeix.
• added support for sigmoid activation function for fused ops (#4948).

Bug fixes

• webgpu: Fix a potential bug in vector*matrix (#5165). Thanks, @qjia7.
• webgpu: Fix the reduce shader key (#5148). Thanks, @qjia7.
• webgpu: Fix the mirrorPad error (#5131). Thanks, @qjia7.
• webgpu: Add CPU_HANDOFF_SIZE_THRESHOLD (#5119). Thanks, @qjia7.

Performance

• webgpu: Optimize Conv2DBackpropInput (#5155). Thanks, @qjia7.
• webgpu: Add vector*matrix kernel (#5129). Thanks, @qjia7.
• webgpu: Optimize getSamplerAtOutputCoords (#5105). Thanks, @qjia7.
• [webgpu] Use uniform instead of constant for shape (#4819). Thanks, @axinging.

Security

• Fix CVE-2021-23362 in tfjs-backend-webgpu (#5086). Thanks, @aravindvnair99.

Misc

• Bump version to 0.0.1-alpha.7 (#5176).
• Upgrade webgpu to 0.0.1-alpha.6 (#5175).
• [webgpu] Fix string comparison fail and add equal (#5149). Thanks, @axinging.
• webgpu: Fix the OOB error in transform/slice/stridedSlice (#5164). Thanks, @qjia7.
• webgpu: Add Pow kernel (#5130). Thanks, @qjia7.
• [webgpu] Add op StringNGrams (#5118). Thanks, @axinging.
• [webgpu] Fix nan errors (#5121). Thanks, @axinging.
• [webgpu] Update yarn.lock for package types/long (#5125). Thanks, @axinging.
• [webgpu] Add shared memory flag to binary op shared shader key (#5083). Thanks, @axinging.
• webgpu: Remove cpu backend (#5101). Thanks, @qjia7.
• webgpu: Add Conv2DBackpropInput kernel (#5104). Thanks, @qjia7.
• [webgpu] Fix invalid WGPUBindingType (#5082). Thanks, @axinging.
• webgpu: run concat on cpu if possible (#5076). Thanks, @qjia7.
• [webgpu] Reuse GPUBindGroupLayout and GPUPipelineLayout (#5001). Thanks, @axinging.
• webgpu: fix/suppress some failure cases (#5065). Thanks, @qjia7.
• webgpu: Add Transform kernel (#5061). Thanks, @qjia7.
• webgpu: upgrade @webgpu/types (#5062). Thanks, @qjia7.
• [webgpu] Simplify shader key of slice and cropAndResize (#4993). Thanks, @axinging.
• [WebGPU] Expand texture usage for fromPixels (#5016). Thanks, @shaoboyan.
• webgpu: Use outShapeStrides in more places (#4967). Thanks, @qjia7.
• webgpu: Use uniforms for PadV2 paddings (#4966). Thanks, @qjia7.
• [webgpu] Use uniform for program batchNorm, clip, im2col (#4958). Thanks, @axinging.