Preparing search index...
The search index is not available
webgpu-torch
webgpu-torch
Device
Class Device
Abstract
Hierarchy
Device
Index
Constructors
constructor
Properties
_heap
Enabled
_heap
Finalizers
_heaps
_id
_kernels
_type
Accessors
id
type
workgroup
Max
Count
workgroup
Max
Size
Methods
alloc
alloc
Buffer
Heap
create
Heap
Storage
create
Kernel
get
Kernel
heap
Alloc
init
Storage
Constructors
constructor
new
Device
(
id
,
type
)
:
Device
Parameters
id
:
string
type
:
DeviceType
Returns
Device
Properties
Private
_heap
Enabled
_heap
Enabled
:
boolean
= false
Private
_heap
Finalizers
_heap
Finalizers
:
FinalizationRegistry
<
HeapBuffer
<
ArrayBuffer
|
GPUBuffer
>
>
Private
_heaps
_heaps
:
BufferHeap
<
ArrayBuffer
|
GPUBuffer
>
[]
= []
Private
_id
_id
:
string
Private
_kernels
_kernels
:
{
[
key
:
KernelKey
]
:
Kernel
;
}
= {}
Type declaration
[
key
:
KernelKey
]:
Kernel
Private
_type
_type
:
DeviceType
Accessors
id
get
id
(
)
:
string
Returns
string
type
get
type
(
)
:
DeviceType
Returns
DeviceType
Abstract
workgroup
Max
Count
get
workgroupMaxCount
(
)
:
number
Returns
number
Abstract
workgroup
Max
Size
get
workgroupMaxSize
(
)
:
[
number
,
number
,
number
]
Returns
[
number
,
number
,
number
]
Methods
Abstract
alloc
alloc
(
byteSize
)
:
UntypedStorage
Parameters
byteSize
:
number
Returns
UntypedStorage
Abstract
alloc
Buffer
Heap
alloc
Buffer
Heap
(
)
:
BufferHeap
<
ArrayBuffer
|
GPUBuffer
>
Returns
BufferHeap
<
ArrayBuffer
|
GPUBuffer
>
Abstract
create
Heap
Storage
create
Heap
Storage
(
buffer
)
:
UntypedStorage
Parameters
buffer
:
HeapBuffer
<
ArrayBuffer
|
GPUBuffer
>
Returns
UntypedStorage
Abstract
create
Kernel
create
Kernel
(
spec
,
config
)
:
Kernel
Parameters
spec
:
KernelSpec
config
:
KernelConfig
Returns
Kernel
get
Kernel
get
Kernel
(
name
,
config
)
:
Kernel
Parameters
name
:
string
config
:
KernelConfigInput
Returns
Kernel
heap
Alloc
heap
Alloc
(
byteSize
)
:
UntypedStorage
Parameters
byteSize
:
number
Returns
UntypedStorage
Abstract
init
Storage
init
Storage
(
shape
,
dtype
,
init
)
:
UntypedStorage
Parameters
shape
:
Shape
dtype
:
Dtype
init
:
(
(
array
)
=>
void
)
(
array
)
:
void
Parameters
array
:
ATypedArray
Returns
void
Returns
UntypedStorage
Settings
Member Visibility
Protected
Private
Inherited
External
Theme
OS
Light
Dark
On This Page
constructor
_heap
Enabled
_heap
Finalizers
_heaps
_id
_kernels
_type
id
type
workgroup
Max
Count
workgroup
Max
Size
alloc
alloc
Buffer
Heap
create
Heap
Storage
create
Kernel
get
Kernel
heap
Alloc
init
Storage
webgpu-
torch
init
Linear
Function
Nonlinear
Function
calculate
Fan
In
And
Fan
Out
kaiming
Uniform_
uniform_
nn
applications
DeepSDF
diffusers
Timestep
Embed
Sequential
UNet
Model
UNet
Model
Config
Avg
Pooling2d
Container
Conv2d
Conv
Transpose2d
Group
Norm
Linear
Module
Module
List
Parameter
ReLU
Sequential
SiLU
Sigmoid
Tanh
State
Dict
Device
Gradient
Context
Kernel
Tensor
Untyped
Storage
Kernel
Config
Spec
Kernel
Input
Spec
Kernel
Output
Spec
Kernel
Param
Spec
Kernel
Spec
Strided
Shape
ATyped
Array
Compiled
Expr
Device
Id
Device
Type
Deviceish
Dtype
Eval
Env
Expr
Code
Function
Input
Gradient
Function
Gradient
Function
Output
Kernel
Config
Kernel
Config
Input
Kernel
Key
Kernel
Params
Input
Memory
Format
Op
Spec
Op
Type
Shader
Type
Shape
Shapeish
Strides
Tensor
Array
Data
Tensor
Data
Tensor
Spec
abs
absolute
acos
acosh
add
all
any
arccos
arccosh
arcsin
arcsinh
arctan
arctan2
asin
asinh
atan
atan2
broadcast
Batched
Matmul
broadcast
Shapes
canonicalize
Dim
cat
ceil
check
clone
contiguous
Strided
Shape
conv2d
copysign
cos
cosh
count
Nonzero
default
Strides
deg2rad
div
divide
empty
exp
exp2
expm1
fix
flatten
floor
frac
gather
get
Shape
has
WebGPU
hypot
init
WebGPUAsync
ldexp
linear
log
log10
log1p
log2
logaddexp
logaddexp2
matmul
mean
mm
mul
multiply
neg
negative
norm
numel
ones
positive
pow
prod
rad2deg
reciprocal
relu
reshape
reshape
Batched
Matmul
round
rsqrt
save
Safetensors
Async
shape
Size
shapes
Are
Equal
sigmoid
sign
silu
sin
sinc
sinh
sqrt
square
squeeze
strided
Shape
Is
Contiguous
sub
subtract
sum
t
tan
tanh
tensor
trunc
unsqueeze
validate
Dim
Length
validate
Idx
view
xlogy
zeros
Generated using
TypeDoc