@tensorflow/tfjs-backend-webgpu

v4.21.0
This package adds a GPU accelerated [WebGPU](https://www.w3.org/TR/webgpu/) backend to TensorFlow.js. It currently supports the following models: - BlazeFace - BodyPix - Face landmarks detection - HandPose - MobileNet - PoseDetection - Universal sentence

Readme Versions Dependencies Scripts

test

yarn --cwd .. bazel test tfjs-backend-webgpu:tfjs-backend-webgpu_test --test_output=streamed

build

yarn --cwd .. bazel build tfjs-backend-webgpu:tfjs-backend-webgpu_pkg

bundle

yarn build

build-ci

yarn build

test-dev

yarn --cwd .. bazel run tfjs-backend-webgpu:tfjs-backend-webgpu_test --test_output=streamed

benchmark

yarn test --//:grep=benchmark

build-npm

yarn build

bundle-ci

yarn bundle

publish-npm

yarn --cwd .. bazel run tfjs-backend-webgpu:tfjs-backend-webgpu_pkg.publish
npm i @tensorflow/[email protected]

Metadata

Downloads