init nn

Device GradientContext Kernel Tensor UntypedStorage

KernelConfigSpec KernelInputSpec KernelOutputSpec KernelParamSpec KernelSpec StridedShape

ATypedArray CompiledExpr DeviceId DeviceType Deviceish Dtype EvalEnv ExprCode FunctionInput GradientFunction GradientFunctionOutput KernelConfig KernelConfigInput KernelKey KernelParamsInput MemoryFormat OpSpec OpType ShaderType Shape Shapeish Strides TensorArrayData TensorData TensorSpec

abs absolute acos acosh add all any arccos arccosh arcsin arcsinh arctan arctan2 asin asinh atan atan2 broadcastBatchedMatmul broadcastShapes canonicalizeDim cat ceil check clone contiguousStridedShape conv2d copysign cos cosh countNonzero defaultStrides deg2rad div divide empty exp exp2 expm1 fix flatten floor frac gather getShape hasWebGPU hypot initWebGPUAsync 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 reshapeBatchedMatmul round rsqrt saveSafetensorsAsync shapeSize shapesAreEqual sigmoid sign silu sin sinc sinh sqrt square squeeze stridedShapeIsContiguous sub subtract sum t tan tanh tensor trunc unsqueeze validateDimLength validateIdx view xlogy zeros