tinygrab
Contents:
tinygrab
tinygrab
Index
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Y
|
Z
_
__del__() (tinygrad.runtime.ops_gpu.CLProgram method)
__deletable__ (tinygrad.lazy.LazyBuffer attribute)
(tinygrad.tensor.Tensor attribute)
__floordiv__ (tinygrad.shape.symbolic.SumNode attribute)
__mul__ (tinygrad.shape.symbolic.SumNode attribute)
__slots__ (tinygrad.tensor.Tensor attribute)
_alloc (tinygrad.runtime.ops_cpu.NumpyAllocator attribute)
_max (tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
_min (tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
A
a (tinygrad.shape.symbolic.DivNode attribute)
(tinygrad.shape.symbolic.LtNode attribute)
(tinygrad.shape.symbolic.ModNode attribute)
(tinygrad.shape.symbolic.MulNode attribute)
(tinygrad.shape.symbolic.OpNode attribute)
(tinygrad.tensor.Tensor attribute)
,
[1]
abs() (tinygrad.tensor.Tensor method)
acc (tinygrad.codegen.linearizer.Linearizer attribute)
acc_offsets() (tinygrad.codegen.kernel.Kernel method)
acosh() (tinygrad.tensor.Tensor method)
adam (tinygrad.nn.optim.LAMB attribute)
Adam() (in module tinygrad.nn.optim)
AdamW() (in module tinygrad.nn.optim)
Add (class in tinygrad.mlops)
ADD (tinygrad.ops.BinaryOps attribute)
add() (tinygrad.tensor.Tensor method)
add_st_node() (in module tinygrad.graph)
alias_buffer() (tinygrad.codegen.kernel.Kernel method)
all_int() (in module tinygrad.helpers)
all_same() (in module tinygrad.helpers)
alloc() (tinygrad.device.Allocator method)
(tinygrad.device.LRUAllocator method)
alloc_if_needed() (tinygrad.jit.PlaceHolder method)
Allocator (class in tinygrad.device)
allocator (tinygrad.device.Interpreted attribute)
Allocator (tinygrad.runtime.ops_webgpu.WebGpuAllocator attribute)
allow_test_size (in module tinygrad.features.search)
,
[1]
ALU (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
amt (in module tinygrad.features.search)
(tinygrad.codegen.kernel.Opt attribute)
,
[1]
AndNode (class in tinygrad.shape.symbolic)
ands() (tinygrad.shape.symbolic.Node static method)
ansilen (tinygrad.codegen.kernel.Kernel attribute)
ansilen() (in module tinygrad.helpers)
ansistrip() (in module tinygrad.helpers)
applied_opts (tinygrad.codegen.kernel.Kernel attribute)
,
[1]
(tinygrad.codegen.linearizer.Linearizer.self attribute)
applied_opts_cache (tinygrad.codegen.kernel.Kernel attribute)
(tinygrad.codegen.linearizer.Linearizer.self attribute)
apply() (tinygrad.tensor.Function class method)
apply_opt() (tinygrad.codegen.kernel.Kernel method)
apply_tensor_cores() (tinygrad.codegen.kernel.Kernel method)
arange() (tinygrad.tensor.Tensor static method)
arch (tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
arg (in module tinygrad.runtime.ops_torch)
(tinygrad.codegen.linearizer.Linearizer attribute)
(tinygrad.codegen.linearizer.UOp attribute)
,
[1]
(tinygrad.lazy.LazyBuffer attribute)
,
[1]
,
[2]
,
[3]
(tinygrad.mlops.Flip attribute)
(tinygrad.ops.LazyOp attribute)
,
[1]
(tinygrad.shape.view.View attribute)
,
[1]
(tinygrad.tensor.Tensor attribute)
arg_int_prefix (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
argfix() (in module tinygrad.helpers)
argmax() (tinygrad.tensor.Tensor method)
argmin() (tinygrad.tensor.Tensor method)
args (in module tinygrad.renderer.llvmir)
argsort() (in module tinygrad.helpers)
as_buffer (tinygrad.runtime.ops_cpu.NumpyAllocator attribute)
as_buffer() (tinygrad.runtime.ops_cpu.NumpyAllocator method)
(tinygrad.runtime.ops_disk.DiskAllocator method)
AS_STRIDED (tinygrad.ops.MovementOps attribute)
,
[1]
as_strided() (in module tinygrad.runtime.ops_torch)
(tinygrad.runtime.ops_disk.DiskBuffer method)
asinh() (tinygrad.tensor.Tensor method)
assign() (tinygrad.tensor.Tensor method)
ast (in module tinygrad.lazy)
(in module tinygrad.ops)
(tinygrad.codegen.kernel.Kernel attribute)
(tinygrad.ops.ScheduleItem attribute)
ast_parse() (tinygrad.codegen.linearizer.Linearizer method)
atanh() (tinygrad.tensor.Tensor method)
avg_pool2d() (tinygrad.tensor.Tensor method)
ax1 (tinygrad.tensor.Tensor attribute)
ax2 (tinygrad.tensor.Tensor attribute)
axis (tinygrad.codegen.kernel.Opt attribute)
,
[1]
(tinygrad.tensor.Tensor attribute)
axis_is_masked() (tinygrad.shape.shapetracker.ShapeTracker method)
B
b (tinygrad.shape.symbolic.DivNode attribute)
(tinygrad.shape.symbolic.LtNode attribute)
(tinygrad.shape.symbolic.ModNode attribute)
(tinygrad.shape.symbolic.MulNode attribute)
(tinygrad.shape.symbolic.Node attribute)
,
[1]
(tinygrad.shape.symbolic.NumNode attribute)
(tinygrad.shape.symbolic.OpNode attribute)
b1 (in module tinygrad.nn.optim)
,
[1]
(tinygrad.nn.optim.LAMB attribute)
b2 (in module tinygrad.nn.optim)
,
[1]
(tinygrad.nn.optim.LAMB attribute)
backward (tinygrad.mlops.Cast attribute)
(tinygrad.mlops.Contiguous attribute)
(tinygrad.mlops.ContiguousBackward attribute)
(tinygrad.mlops.Neg attribute)
(tinygrad.mlops.Zero attribute)
backward() (tinygrad.mlops.Add method)
(tinygrad.mlops.Cast method)
(tinygrad.mlops.Contiguous method)
(tinygrad.mlops.ContiguousBackward method)
(tinygrad.mlops.Div method)
(tinygrad.mlops.Exp method)
(tinygrad.mlops.Expand method)
,
[1]
(tinygrad.mlops.Flip method)
(tinygrad.mlops.Log method)
(tinygrad.mlops.Max method)
(tinygrad.mlops.Mul method)
(tinygrad.mlops.Neg method)
(tinygrad.mlops.Pad method)
(tinygrad.mlops.Permute method)
(tinygrad.mlops.Relu method)
(tinygrad.mlops.Reshape method)
(tinygrad.mlops.Shrink method)
(tinygrad.mlops.Sigmoid method)
(tinygrad.mlops.Sin method)
(tinygrad.mlops.Sqrt method)
(tinygrad.mlops.Sub method)
(tinygrad.mlops.Sum method)
(tinygrad.mlops.Where method)
(tinygrad.mlops.Zero method)
(tinygrad.tensor.Function method)
(tinygrad.tensor.Tensor method)
BARRIER (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
barrier (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
base (tinygrad.lazy.LazyBuffer attribute)
(tinygrad.lazy.LazyBuffer property)
base_bufs (tinygrad.lazy.LazyBuffer attribute)
base_shape (in module tinygrad.features.image)
batchnorm() (tinygrad.tensor.Tensor method)
bb (in module tinygrad.renderer.llvmir)
beam_search() (in module tinygrad.features.search)
bfloat16 (tinygrad.helpers.dtypes attribute)
bias (in module tinygrad.features.image)
(tinygrad.tensor.Tensor attribute)
binary_crossentropy() (tinygrad.tensor.Tensor method)
binary_crossentropy_logits() (tinygrad.tensor.Tensor method)
BinaryOps (class in tinygrad.ops)
bind() (tinygrad.shape.symbolic.NumNode method)
(tinygrad.shape.symbolic.Variable method)
bitcast (tinygrad.lazy.LazyBuffer attribute)
bitcast() (tinygrad.tensor.Tensor method)
bool (tinygrad.helpers.dtypes attribute)
buf_dtype (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
buf_index (in module tinygrad.renderer.llvmir)
buf_name (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
buf_to_dtype (in module tinygrad.renderer.llvmir)
buf_uops (tinygrad.codegen.linearizer.Linearizer.self attribute)
Buffer (class in tinygrad.device)
buffer_prefix (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
buffer_suffix (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
BufferOps (class in tinygrad.ops)
buffers (tinygrad.lazy.LazyBuffer property)
(tinygrad.nn.optim.Optimizer attribute)
(tinygrad.ops.LazyOp property)
bufs (tinygrad.codegen.kernel.Kernel attribute)
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
bufs_from_lin() (in module tinygrad.features.search)
build() (tinygrad.device.CompiledASTRunner method)
build_kernel_node_params() (tinygrad.features.graph.cuda.CUDAGraph method)
C
cachable (tinygrad.codegen.linearizer.Linearizer attribute)
CACHEDB (in module tinygrad.helpers)
CACHELEVEL (in module tinygrad.helpers)
Cast (class in tinygrad.mlops)
cast (tinygrad.codegen.linearizer.Linearizer attribute)
CAST (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
(tinygrad.ops.UnaryOps attribute)
cast() (in module tinygrad.renderer.llvmir)
(tinygrad.codegen.linearizer.Linearizer method)
(tinygrad.lazy.LazyBuffer method)
(tinygrad.runtime.ops_disk.DiskBuffer method)
(tinygrad.tensor.Tensor method)
cat() (tinygrad.tensor.Tensor method)
cb (in module tinygrad.runtime.ops_cuda)
cdf (tinygrad.tensor.Tensor attribute)
ceil() (tinygrad.tensor.Tensor method)
celu() (tinygrad.tensor.Tensor method)
check() (in module tinygrad.runtime.ops_cuda)
(in module tinygrad.runtime.ops_gpu)
checked() (in module tinygrad.runtime.ops_gpu)
children (tinygrad.lazy.LazyBuffer attribute)
(tinygrad.ops.LazyOp property)
chunk() (tinygrad.tensor.Tensor method)
CLAllocator (class in tinygrad.runtime.ops_gpu)
clang() (tinygrad.tensor.Tensor method)
ClangProgram (class in tinygrad.runtime.ops_clang)
CLDevice (class in tinygrad.runtime.ops_gpu)
clear_l2 (in module tinygrad.features.search)
clip (tinygrad.tensor.Tensor attribute)
clip() (tinygrad.tensor.Tensor method)
clprg (in module tinygrad.features.search)
CLProgram (class in tinygrad.runtime.ops_gpu)
CMPLT (tinygrad.ops.BinaryOps attribute)
cnt (in module tinygrad.features.search)
(tinygrad.jit.TinyJit attribute)
code_for_op (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
codegen (tinygrad.device.Interpreted attribute)
colored() (in module tinygrad.helpers)
colored_shape() (tinygrad.codegen.kernel.Kernel method)
colors (tinygrad.codegen.kernel.Kernel.self attribute)
colors() (tinygrad.codegen.kernel.Kernel method)
compile_cuda_style() (in module tinygrad.helpers)
Compiled (class in tinygrad.device)
(tinygrad.runtime.ops_webgpu.WebGpuDevice attribute)
CompiledASTRunner (class in tinygrad.device)
compiler_context (tinygrad.runtime.ops_gpu.CLDevice attribute)
,
[1]
cond (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
const (tinygrad.codegen.linearizer.Linearizer attribute)
CONST (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
(tinygrad.ops.BufferOps attribute)
,
[1]
(tinygrad.ops.LoadOps attribute)
,
[1]
const() (in module tinygrad.renderer.llvmir)
(tinygrad.codegen.linearizer.Linearizer method)
(tinygrad.lazy.LazyBuffer method)
ConstBuffer (class in tinygrad.ops)
consume_flops() (tinygrad.ops.FlopCounter method)
Context (class in tinygrad.helpers)
ContextVar (class in tinygrad.helpers)
Contiguous (class in tinygrad.mlops)
CONTIGUOUS (tinygrad.ops.LoadOps attribute)
,
[1]
contiguous (tinygrad.shape.shapetracker.ShapeTracker property)
(tinygrad.shape.view.View attribute)
,
[1]
contiguous() (tinygrad.lazy.LazyBuffer method)
(tinygrad.tensor.Tensor method)
contiguous_backward() (tinygrad.tensor.Tensor method)
ContiguousBackward (class in tinygrad.mlops)
conv2d() (tinygrad.tensor.Tensor method)
conv_transpose2d() (tinygrad.tensor.Tensor method)
copy() (tinygrad.codegen.kernel.Kernel method)
copy_to_device() (tinygrad.lazy.LazyBuffer method)
copyin (tinygrad.runtime.ops_cpu.NumpyAllocator attribute)
copyin() (tinygrad.device.Allocator method)
(tinygrad.device.Buffer method)
(tinygrad.runtime.ops_cpu.NumpyAllocator method)
(tinygrad.runtime.ops_cuda.CUDAAllocator method)
(tinygrad.runtime.ops_disk.DiskAllocator method)
(tinygrad.runtime.ops_gpu.CLAllocator method)
(tinygrad.runtime.ops_torch.TorchAllocator method)
(tinygrad.runtime.ops_webgpu.WebGpuAllocator method)
copyout (tinygrad.runtime.ops_cpu.NumpyAllocator attribute)
copyout() (tinygrad.device.Allocator method)
(tinygrad.runtime.ops_cpu.NumpyAllocator method)
(tinygrad.runtime.ops_cuda.CUDAAllocator method)
(tinygrad.runtime.ops_disk.DiskAllocator method)
(tinygrad.runtime.ops_gpu.CLAllocator method)
(tinygrad.runtime.ops_torch.TorchAllocator method)
(tinygrad.runtime.ops_webgpu.WebGpuAllocator method)
corealize() (tinygrad.tensor.Tensor static method)
cos() (tinygrad.tensor.Tensor method)
cosh() (tinygrad.tensor.Tensor method)
cpu() (tinygrad.tensor.Tensor method)
cpu_time_execution() (in module tinygrad.helpers)
create() (tinygrad.shape.view.View static method)
create_lazybuffer() (in module tinygrad.lazy)
create_node() (in module tinygrad.shape.symbolic)
create_rednode() (in module tinygrad.shape.symbolic)
CStyleLanguage (class in tinygrad.renderer.cstyle)
ctx (tinygrad.shape.symbolic.Node attribute)
cu_time_execution() (in module tinygrad.runtime.ops_cuda)
cuda() (tinygrad.tensor.Tensor method)
CUDAAllocator (class in tinygrad.runtime.ops_cuda)
CUDADevice (class in tinygrad.runtime.ops_cuda)
CUDAGraph (class in tinygrad.features.graph.cuda)
CUDALanguage (class in tinygrad.renderer.cstyle)
CUDAProgram (class in tinygrad.runtime.ops_cuda)
CUDARenderer() (in module tinygrad.renderer.cstyle)
cumsum() (tinygrad.tensor.Tensor method)
CUSTOM (tinygrad.ops.LoadOps attribute)
,
[1]
custom_random() (in module tinygrad.tensor)
CustomOp (class in tinygrad.realize)
D
db_connection() (in module tinygrad.helpers)
dedup() (in module tinygrad.helpers)
deepwalk() (tinygrad.tensor.Tensor method)
default_arch_name (tinygrad.runtime.ops_cuda.CUDADevice attribute)
,
[1]
default_type (tinygrad.tensor.Tensor attribute)
,
[1]
DEFINE_ACC (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
DEFINE_GLOBAL (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
DEFINE_LOCAL (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
detach() (tinygrad.tensor.Tensor method)
device (tinygrad.codegen.kernel.LinearizerOptions attribute)
,
[1]
(tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
(tinygrad.lazy.LazyBuffer attribute)
,
[1]
,
[2]
(tinygrad.nn.optim.Optimizer attribute)
(tinygrad.runtime.ops_cuda.CUDAAllocator attribute)
(tinygrad.runtime.ops_cuda.CUDAProgram attribute)
(tinygrad.runtime.ops_disk.DiskAllocator attribute)
(tinygrad.runtime.ops_gpu.CLAllocator attribute)
(tinygrad.runtime.ops_gpu.CLProgram attribute)
(tinygrad.tensor.Function attribute)
(tinygrad.tensor.Tensor property)
device_ids (tinygrad.runtime.ops_gpu.CLDevice attribute)
,
[1]
Dict (in module tinygrad.lazy)
dilation (in module tinygrad.features.image)
(tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
dim (tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
,
[3]
,
[4]
dims (tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
disable_cache (in module tinygrad.features.search)
disallow_bool (in module tinygrad.runtime.ops_torch)
disk() (tinygrad.tensor.Tensor method)
disk_fxn_for_op (tinygrad.runtime.ops_disk.DiskDevice attribute)
DiskAllocator (class in tinygrad.runtime.ops_disk)
(tinygrad.runtime.ops_disk.DiskDevice attribute)
DiskBuffer (class in tinygrad.runtime.ops_disk)
diskcache() (in module tinygrad.helpers)
diskcache_get() (in module tinygrad.helpers)
diskcache_put() (in module tinygrad.helpers)
DiskDevice (class in tinygrad.runtime.ops_disk)
Div (class in tinygrad.mlops)
DIV (tinygrad.ops.BinaryOps attribute)
div() (tinygrad.tensor.Tensor method)
DivNode (class in tinygrad.shape.symbolic)
do_reduce (tinygrad.codegen.linearizer.Linearizer attribute)
dont_use_locals (tinygrad.codegen.kernel.Kernel attribute)
,
[1]
dot() (tinygrad.tensor.Tensor method)
double (tinygrad.helpers.dtypes attribute)
dropout() (tinygrad.tensor.Tensor method)
DType (class in tinygrad.helpers)
dtype (in module tinygrad.renderer.llvmir)
(tinygrad.codegen.kernel.LocalBuffer attribute)
,
[1]
(tinygrad.codegen.linearizer.Linearizer attribute)
(tinygrad.codegen.linearizer.UOp attribute)
,
[1]
(tinygrad.lazy.LazyBuffer attribute)
,
[1]
,
[2]
(tinygrad.lazy.LazyBuffer property)
(tinygrad.ops.ConstBuffer attribute)
(tinygrad.ops.FlopCounter attribute)
,
[1]
(tinygrad.ops.MemBuffer attribute)
(tinygrad.runtime.ops_disk.DiskBuffer attribute)
(tinygrad.tensor.Tensor attribute)
(tinygrad.tensor.Tensor property)
dtype_in (tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
dtype_out (tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
dtypes (class in tinygrad.helpers)
E
e() (tinygrad.lazy.LazyBuffer method)
earlybufs (tinygrad.codegen.kernel.Kernel attribute)
einsum (in module tinygrad.runtime.ops_cpu)
einsum_mulacc() (in module tinygrad.runtime.ops_cpu)
element_size() (tinygrad.tensor.Tensor method)
elu() (tinygrad.tensor.Tensor method)
EMPTY (tinygrad.ops.LoadOps attribute)
,
[1]
empty() (tinygrad.tensor.Tensor static method)
enable (in module tinygrad.runtime.ops_cuda)
enabled (tinygrad.helpers.Profiling attribute)
(tinygrad.helpers.Timing attribute)
encode_args_cuda_style() (in module tinygrad.helpers)
encode_args_info() (tinygrad.features.graph.cuda.CUDAGraph method)
END (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
engine (tinygrad.runtime.ops_llvm.LLVM attribute)
,
[1]
eps (in module tinygrad.nn.optim)
,
[1]
(tinygrad.nn.optim.LAMB attribute)
exec() (tinygrad.device.JITRunner method)
Exp (class in tinygrad.mlops)
exp (tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
,
[3]
exp() (tinygrad.tensor.Tensor method)
EXP2 (tinygrad.ops.UnaryOps attribute)
exp2() (tinygrad.tensor.Tensor method)
Expand (class in tinygrad.mlops)
expand (in module tinygrad.runtime.ops_cpu)
EXPAND (tinygrad.ops.MovementOps attribute)
,
[1]
expand() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
(tinygrad.shape.shapetracker.ShapeTracker method)
(tinygrad.shape.symbolic.Node method)
(tinygrad.shape.view.View method)
(tinygrad.tensor.Tensor method)
expand_idx() (tinygrad.shape.symbolic.Node method)
expected_name_sts_dtype (tinygrad.jit.TinyJit attribute)
expected_vals (tinygrad.jit.TinyJit attribute)
expr (tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
(tinygrad.shape.symbolic.Variable attribute)
expr_idxs() (in module tinygrad.shape.shapetracker)
(tinygrad.shape.shapetracker.ShapeTracker method)
expr_node() (in module tinygrad.shape.shapetracker)
(tinygrad.shape.shapetracker.ShapeTracker method)
expr_node_mask() (in module tinygrad.shape.shapetracker)
external_local_bufs (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
extra_args (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
extra_opts (tinygrad.codegen.kernel.Kernel attribute)
eye() (tinygrad.tensor.Tensor static method)
F
fd (tinygrad.runtime.ops_disk.UnderlyingDiskBuffer attribute)
fetch() (in module tinygrad.helpers)
fields() (tinygrad.helpers.dtypes static method)
fill_value (tinygrad.tensor.Tensor attribute)
filter_strides() (in module tinygrad.shape.view)
first_reduce (tinygrad.codegen.kernel.Kernel attribute)
,
[1]
(tinygrad.codegen.kernel.Kernel property)
(tinygrad.codegen.kernel.Kernel.self attribute)
,
[1]
flat_components (tinygrad.shape.symbolic.SumNode property)
flat_mv() (in module tinygrad.helpers)
flatten() (in module tinygrad.helpers)
(tinygrad.tensor.Tensor method)
Flip (class in tinygrad.mlops)
flip() (tinygrad.tensor.Tensor method)
float (tinygrad.helpers.dtypes attribute)
float() (tinygrad.tensor.Tensor method)
float16 (tinygrad.helpers.dtypes attribute)
float32 (tinygrad.helpers.dtypes attribute)
float4 (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
float4_axis() (tinygrad.codegen.kernel.Kernel method)
float64 (tinygrad.helpers.dtypes attribute)
floor() (tinygrad.tensor.Tensor method)
FlopCounter (class in tinygrad.ops)
flops (tinygrad.ops.FlopCounter attribute)
,
[1]
forward (tinygrad.mlops.Cast attribute)
(tinygrad.mlops.Contiguous attribute)
(tinygrad.mlops.ContiguousBackward attribute)
(tinygrad.mlops.Neg attribute)
(tinygrad.mlops.Zero attribute)
forward() (tinygrad.mlops.Add method)
(tinygrad.mlops.Cast method)
(tinygrad.mlops.Contiguous method)
(tinygrad.mlops.ContiguousBackward method)
(tinygrad.mlops.Div method)
(tinygrad.mlops.Exp method)
(tinygrad.mlops.Expand method)
,
[1]
(tinygrad.mlops.Flip method)
(tinygrad.mlops.Less method)
(tinygrad.mlops.Log method)
(tinygrad.mlops.Max method)
(tinygrad.mlops.Mul method)
(tinygrad.mlops.Neg method)
(tinygrad.mlops.Pad method)
(tinygrad.mlops.Permute method)
(tinygrad.mlops.Relu method)
(tinygrad.mlops.Reshape method)
(tinygrad.mlops.Shrink method)
(tinygrad.mlops.Sigmoid method)
(tinygrad.mlops.Sin method)
(tinygrad.mlops.Sqrt method)
(tinygrad.mlops.Sub method)
(tinygrad.mlops.Sum method)
(tinygrad.mlops.Where method)
(tinygrad.mlops.Zero method)
(tinygrad.tensor.Function method)
frac (tinygrad.helpers.Profiling attribute)
free() (tinygrad.device.Allocator method)
(tinygrad.device.LRUAllocator method)
free_cache() (tinygrad.device.LRUAllocator method)
FROM (tinygrad.ops.LoadOps attribute)
,
[1]
from_mv() (in module tinygrad.helpers)
from_np() (tinygrad.helpers.dtypes static method)
from_shape() (tinygrad.shape.shapetracker.ShapeTracker static method)
fromCPU() (tinygrad.device.Buffer static method)
(tinygrad.lazy.LazyBuffer static method)
fromimport() (in module tinygrad.helpers)
full() (tinygrad.tensor.Tensor static method)
full_buf_index (tinygrad.codegen.kernel.Kernel attribute)
(tinygrad.codegen.kernel.Kernel.self attribute)
full_like() (tinygrad.tensor.Tensor method)
full_shape (tinygrad.codegen.kernel.Kernel attribute)
(tinygrad.codegen.kernel.Kernel property)
(tinygrad.codegen.kernel.Kernel.self attribute)
,
[1]
,
[2]
full_unupcasted_shape (tinygrad.codegen.kernel.Kernel property)
func (in module tinygrad.renderer.llvmir)
func_dtypes (in module tinygrad.renderer.llvmir)
Function (class in tinygrad.tensor)
function_name (tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
fxn (tinygrad.device.InterpretedASTRunner attribute)
(tinygrad.jit.TinyJit attribute)
(tinygrad.realize.CustomOp attribute)
(tinygrad.runtime.ops_clang.ClangProgram attribute)
fxn_for_op (tinygrad.device.Interpreted attribute)
G
gather() (tinygrad.tensor.Tensor method)
gelu() (tinygrad.tensor.Tensor method)
generic_var_prefix (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
GEP (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
get_bounds() (tinygrad.shape.symbolic.DivNode method)
(tinygrad.shape.symbolic.LtNode method)
(tinygrad.shape.symbolic.ModNode method)
(tinygrad.shape.symbolic.MulNode method)
(tinygrad.shape.symbolic.OpNode method)
get_bytes() (in module tinygrad.helpers)
get_child() (in module tinygrad.helpers)
get_contraction() (in module tinygrad.shape.shapetracker)
get_grouped_dims() (in module tinygrad.codegen.linearizer)
get_input_replace() (in module tinygrad.jit)
get_jc_idxs_with_updatable_launch_dims() (in module tinygrad.jit)
get_jc_idxs_with_updatable_var_vals() (in module tinygrad.jit)
get_jit_stats() (in module tinygrad.jit)
get_lazyop_info() (in module tinygrad.ops)
get_lazyops() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
,
[1]
get_linearizer() (tinygrad.device.Compiled method)
get_linearizer_actions() (in module tinygrad.features.search)
get_movementroot() (in module tinygrad.lazy)
get_movementroot_contiguous() (in module tinygrad.lazy)
get_parameters() (in module tinygrad.nn.state)
get_reduce_acc (tinygrad.codegen.linearizer.Linearizer attribute)
get_reduce_acc() (tinygrad.codegen.linearizer.Linearizer method)
get_runner() (tinygrad.device.Compiled method)
(tinygrad.device.Interpreted method)
get_sop() (in module tinygrad.graph)
get_state_dict() (in module tinygrad.nn.state)
get_strides (in module tinygrad.runtime.ops_cpu)
get_upcast_dim() (tinygrad.codegen.kernel.Kernel method)
getenv() (in module tinygrad.helpers)
gfr_backup (tinygrad.codegen.linearizer.Linearizer attribute)
gid (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
global_dims (tinygrad.codegen.kernel.Kernel attribute)
(tinygrad.codegen.kernel.Kernel property)
global_load() (tinygrad.codegen.linearizer.Linearizer method)
global_max (tinygrad.codegen.kernel.LinearizerOptions attribute)
,
[1]
(tinygrad.codegen.linearizer.Linearizer.self.opts attribute)
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
global_mem (tinygrad.helpers.GlobalCounters attribute)
global_ops (tinygrad.helpers.GlobalCounters attribute)
global_size (in module tinygrad.features.search)
global_store() (tinygrad.codegen.linearizer.Linearizer method)
GlobalCounters (class in tinygrad.helpers)
glorot_uniform() (tinygrad.tensor.Tensor static method)
gpu() (tinygrad.tensor.Tensor method)
GPUDevice (in module tinygrad.runtime.ops_gpu)
grad (tinygrad.tensor.Tensor attribute)
grad_output (tinygrad.mlops.Add attribute)
(tinygrad.mlops.Sub attribute)
graph (tinygrad.device.Interpreted attribute)
graph_add_kernel_node() (tinygrad.features.graph.cuda.CUDAGraph method)
graph_create() (tinygrad.features.graph.cuda.CUDAGraph method)
graph_exec_kernel_node_set_params() (tinygrad.features.graph.cuda.CUDAGraph method)
graph_instantiate() (tinygrad.features.graph.cuda.CUDAGraph method)
graph_launch() (tinygrad.features.graph.cuda.CUDAGraph method)
graph_uops() (in module tinygrad.graph)
GraphException
GROUP (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
group_for_reduce (tinygrad.codegen.kernel.Kernel attribute)
,
[1]
,
[2]
(tinygrad.codegen.kernel.Kernel.self attribute)
groups (in module tinygrad.features.image)
(tinygrad.tensor.Tensor attribute)
GROUPTOP (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
H
half (tinygrad.helpers.dtypes attribute)
half() (tinygrad.tensor.Tensor method)
half_prekernel (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
hand_coded_optimizations() (tinygrad.codegen.kernel.Kernel method)
hardswish() (tinygrad.tensor.Tensor method)
hardtanh() (tinygrad.tensor.Tensor method)
has_local (tinygrad.codegen.kernel.LinearizerOptions attribute)
,
[1]
has_shared (tinygrad.codegen.kernel.LinearizerOptions attribute)
,
[1]
hash (tinygrad.ops.LazyOp property)
(tinygrad.shape.symbolic.Node property)
high (tinygrad.tensor.Tensor attribute)
,
[1]
hip() (tinygrad.tensor.Tensor method)
HIPLanguage (class in tinygrad.renderer.cstyle)
HIPRenderer() (in module tinygrad.renderer.cstyle)
I
i (tinygrad.codegen.kernel.Kernel attribute)
,
[1]
idx (in module tinygrad.shape.shapetracker)
,
[1]
(tinygrad.ops.MemBuffer attribute)
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
(tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
idxs (tinygrad.shape.shapetracker.ShapeTracker attribute)
idxs_to_idx() (in module tinygrad.shape.shapetracker)
idxy (in module tinygrad.features.image)
IF (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
ignore_valid (tinygrad.shape.shapetracker.ShapeTracker attribute)
image_conv2d() (in module tinygrad.features.image)
image_dot() (in module tinygrad.features.image)
ImageDType (class in tinygrad.helpers)
imagef() (tinygrad.helpers.dtypes static method)
imageh() (tinygrad.helpers.dtypes static method)
include_0 (in module tinygrad.features.search)
indices (tinygrad.tensor.Tensor attribute)
info (tinygrad.codegen.kernel.Kernel attribute)
init_c_struct_t() (in module tinygrad.helpers)
init_c_var() (in module tinygrad.helpers)
input_order (tinygrad.mlops.Permute attribute)
input_rawbuffers (tinygrad.features.graph.cuda.CUDAGraph attribute)
input_replace (tinygrad.jit.TinyJit attribute)
input_shape (tinygrad.mlops.Expand attribute)
(tinygrad.mlops.Reshape attribute)
(tinygrad.mlops.Sum attribute)
input_type (in module tinygrad.renderer.llvmir)
inputs (tinygrad.ops.ScheduleItem attribute)
insert_before (tinygrad.codegen.linearizer.Linearizer attribute)
int (tinygrad.helpers.dtypes attribute)
int16 (tinygrad.helpers.dtypes attribute)
int32 (tinygrad.helpers.dtypes attribute)
int64 (tinygrad.helpers.dtypes attribute)
int8 (tinygrad.helpers.dtypes attribute)
Interpreted (class in tinygrad.device)
InterpretedASTRunner (class in tinygrad.device)
is_bool() (in module tinygrad.renderer.llvmir)
is_float() (tinygrad.helpers.dtypes static method)
is_floating_point() (tinygrad.tensor.Tensor method)
is_int() (tinygrad.helpers.dtypes static method)
is_sym_int() (in module tinygrad.shape.symbolic)
is_unrealized_const() (tinygrad.lazy.LazyBuffer method)
is_unrealized_contiguous_const() (tinygrad.lazy.LazyBuffer method)
is_unsigned() (tinygrad.helpers.dtypes static method)
item() (tinygrad.tensor.Tensor method)
itemsize (tinygrad.helpers.DType attribute)
(tinygrad.helpers.ImageDType attribute)
iter_idxs() (tinygrad.shape.symbolic.Node static method)
J
jit_cache (tinygrad.features.graph.cuda.CUDAGraph attribute)
(tinygrad.jit.TinyJit attribute)
JitItem (class in tinygrad.jit)
JITRunner (class in tinygrad.device)
K
kaiming_normal() (tinygrad.tensor.Tensor static method)
kaiming_uniform() (tinygrad.tensor.Tensor static method)
keepdim (tinygrad.tensor.Tensor attribute)
Kernel (class in tinygrad.codegen.kernel)
kernel (tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
kernel_cnt (tinygrad.codegen.linearizer.Linearizer attribute)
kernel_count (tinygrad.helpers.GlobalCounters attribute)
kernel_prefix (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
kernel_size (tinygrad.tensor.Tensor attribute)
,
[1]
key (tinygrad.shape.symbolic.Node property)
L
LAMB (class in tinygrad.nn.optim)
LASTLOCAL (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
launch_bounds (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
launch_dims() (tinygrad.device.CompiledASTRunner method)
layernorm() (tinygrad.tensor.Tensor method)
LazyBuffer (class in tinygrad.lazy)
(in module tinygrad.lazy)
lazydata (tinygrad.tensor.Tensor attribute)
LazyOp (class in tinygrad.ops)
leakyrelu() (tinygrad.tensor.Tensor method)
lens (in module tinygrad.nn.state)
Less (class in tinygrad.mlops)
lib (tinygrad.runtime.ops_clang.ClangProgram attribute)
(tinygrad.runtime.ops_cuda.CUDAProgram attribute)
(tinygrad.runtime.ops_gpu.CLProgram attribute)
(tinygrad.runtime.ops_llvm.LLVMProgram attribute)
(tinygrad.runtime.ops_webgpu.WebGPUProgram attribute)
lid (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
limit_dims_to_max() (tinygrad.codegen.kernel.Kernel method)
lin (in module tinygrad.features.search)
,
[1]
,
[2]
,
[3]
linear() (tinygrad.tensor.Tensor method)
linearize() (tinygrad.codegen.linearizer.Linearizer method)
Linearizer (class in tinygrad.codegen.linearizer)
LinearizerOptions (class in tinygrad.codegen.kernel)
LLVM (class in tinygrad.runtime.ops_llvm)
llvm() (tinygrad.tensor.Tensor method)
LLVMProgram (class in tinygrad.runtime.ops_llvm)
LOAD (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
(tinygrad.ops.BufferOps attribute)
,
[1]
load_state_dict() (in module tinygrad.nn.state)
loaded_buffers (tinygrad.codegen.linearizer.Linearizer attribute)
loadop() (tinygrad.lazy.LazyBuffer static method)
LoadOps (class in tinygrad.ops)
LOCAL (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
local (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
local_alias (tinygrad.codegen.kernel.Kernel attribute)
local_dims (tinygrad.codegen.kernel.Kernel attribute)
,
[1]
(tinygrad.codegen.kernel.Kernel.self attribute)
local_max (tinygrad.codegen.kernel.LinearizerOptions attribute)
,
[1]
(tinygrad.codegen.linearizer.Linearizer.self.opts attribute)
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
local_size (tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
LocalBuffer (class in tinygrad.codegen.kernel)
Log (class in tinygrad.mlops)
log (tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
log() (tinygrad.tensor.Tensor method)
LOG2 (tinygrad.ops.UnaryOps attribute)
log2() (tinygrad.tensor.Tensor method)
log_schedule_item() (in module tinygrad.graph)
log_softmax() (tinygrad.tensor.Tensor method)
LOOP (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
loop_ctx (tinygrad.codegen.linearizer.Linearizer attribute)
loop_uops (tinygrad.codegen.linearizer.Linearizer.self attribute)
low (tinygrad.tensor.Tensor attribute)
,
[1]
lower_schedule_item() (in module tinygrad.realize)
lr (in module tinygrad.nn.optim)
,
[1]
(tinygrad.nn.optim.LAMB attribute)
(tinygrad.nn.optim.Optimizer attribute)
(tinygrad.nn.optim.SGD attribute)
LRUAllocator (class in tinygrad.device)
LtNode (class in tinygrad.shape.symbolic)
M
m (tinygrad.tensor.Tensor attribute)
make_pair() (in module tinygrad.helpers)
manual_seed() (tinygrad.tensor.Tensor static method)
map_buffers() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
mask (tinygrad.shape.view.View attribute)
,
[1]
match_types() (in module tinygrad.runtime.ops_cpu)
(in module tinygrad.runtime.ops_torch)
matmul() (tinygrad.tensor.Tensor method)
Max (class in tinygrad.mlops)
MAX (tinygrad.ops.BinaryOps attribute)
(tinygrad.ops.ReduceOps attribute)
max (tinygrad.shape.symbolic.Node attribute)
,
[1]
(tinygrad.shape.symbolic.NumNode attribute)
max() (tinygrad.tensor.Tensor method)
max_ (tinygrad.tensor.Tensor attribute)
max_global_size (in module tinygrad.features.search)
max_pool2d() (tinygrad.tensor.Tensor method)
maximum (tinygrad.tensor.Tensor attribute)
maximum() (tinygrad.tensor.Tensor method)
mean (tinygrad.tensor.Tensor attribute)
mean() (tinygrad.tensor.Tensor method)
mem (tinygrad.ops.FlopCounter attribute)
,
[1]
(tinygrad.runtime.ops_disk.UnderlyingDiskBuffer attribute)
mem_cached (tinygrad.helpers.GlobalCounters attribute)
mem_estimate (tinygrad.device.InterpretedASTRunner attribute)
(tinygrad.ops.FlopCounter property)
mem_used (tinygrad.helpers.GlobalCounters attribute)
MemBuffer (class in tinygrad.ops)
membufs (tinygrad.codegen.kernel.Kernel property)
merge_dicts() (in module tinygrad.helpers)
merge_views() (in module tinygrad.shape.shapetracker)
message (tinygrad.jit.GraphException attribute)
metal() (tinygrad.tensor.Tensor method)
MetalLanguage (class in tinygrad.renderer.cstyle)
MetalRenderer() (in module tinygrad.renderer.cstyle)
min (tinygrad.shape.symbolic.Node attribute)
,
[1]
(tinygrad.shape.symbolic.NumNode attribute)
min() (tinygrad.tensor.Tensor method)
min_ (tinygrad.tensor.Tensor attribute)
minimum (tinygrad.tensor.Tensor attribute)
minimum() (tinygrad.tensor.Tensor method)
mish() (tinygrad.tensor.Tensor method)
MOD (tinygrad.ops.BinaryOps attribute)
model (in module tinygrad.nn.state)
ModNode (class in tinygrad.shape.symbolic)
module
tinygrad.codegen.kernel
tinygrad.codegen.linearizer
tinygrad.device
tinygrad.features.graph.cuda
tinygrad.features.image
tinygrad.features.search
tinygrad.graph
tinygrad.helpers
tinygrad.jit
tinygrad.lazy
tinygrad.mlops
tinygrad.nn.optim
tinygrad.nn.state
tinygrad.ops
tinygrad.realize
tinygrad.renderer.cstyle
tinygrad.renderer.llvmir
tinygrad.runtime.ops_clang
tinygrad.runtime.ops_cpu
tinygrad.runtime.ops_cuda
tinygrad.runtime.ops_disk
tinygrad.runtime.ops_gpu
tinygrad.runtime.ops_llvm
tinygrad.runtime.ops_torch
tinygrad.runtime.ops_webgpu
tinygrad.shape.shapetracker
tinygrad.shape.symbolic
tinygrad.shape.view
tinygrad.tensor
module (in module tinygrad.renderer.llvmir)
momentum (tinygrad.nn.optim.SGD attribute)
MOVEMENT_OPS_DISPATCHER (in module tinygrad.lazy)
MovementOps (class in tinygrad.ops)
(in module tinygrad.lazy)
Mul (class in tinygrad.mlops)
MUL (tinygrad.ops.BinaryOps attribute)
mul() (tinygrad.tensor.Tensor method)
MULACC (tinygrad.ops.TernaryOps attribute)
,
[1]
MulNode (class in tinygrad.shape.symbolic)
multinomial() (tinygrad.tensor.Tensor method)
MV_BLOCKSIZE (tinygrad.codegen.kernel.Kernel attribute)
MV_ROWS_PER_THREAD (tinygrad.codegen.kernel.Kernel attribute)
MV_THREADS_PER_ROW (tinygrad.codegen.kernel.Kernel attribute)
N
name (tinygrad.codegen.kernel.LocalBuffer attribute)
,
[1]
(tinygrad.helpers.DType attribute)
(tinygrad.helpers.ImageDType attribute)
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.runtime.ops_clang.ClangProgram attribute)
(tinygrad.runtime.ops_cuda.CUDAProgram attribute)
(tinygrad.runtime.ops_gpu.CLProgram attribute)
(tinygrad.runtime.ops_llvm.LLVMProgram attribute)
(tinygrad.runtime.ops_webgpu.WebGPUProgram attribute)
narg (tinygrad.mlops.Pad attribute)
(tinygrad.mlops.Shrink attribute)
nbytes() (tinygrad.tensor.Tensor method)
ndim (tinygrad.tensor.Tensor property)
needs_input_grad (tinygrad.tensor.Function attribute)
Neg (class in tinygrad.mlops)
NEG (tinygrad.ops.UnaryOps attribute)
neg (tinygrad.tensor.Tensor attribute)
,
[1]
neg() (tinygrad.tensor.Tensor method)
nesterov (tinygrad.nn.optim.SGD attribute)
new_shape (tinygrad.lazy.LazyBuffer attribute)
nm() (in module tinygrad.graph)
nmax (tinygrad.shape.symbolic.Variable attribute)
nmin (tinygrad.shape.symbolic.Variable attribute)
no_grad (tinygrad.tensor.Tensor attribute)
,
[1]
Node (class in tinygrad.shape.symbolic)
nodes (tinygrad.shape.symbolic.Node attribute)
(tinygrad.shape.symbolic.RedNode attribute)
NOLOCALS (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
None (tinygrad.device.Allocator attribute)
normal() (tinygrad.tensor.Tensor static method)
np (tinygrad.helpers.DType attribute)
(tinygrad.helpers.ImageDType attribute)
num (tinygrad.tensor.Tensor attribute)
numel() (tinygrad.tensor.Tensor method)
NumNode (class in tinygrad.shape.symbolic)
numpy() (tinygrad.tensor.Tensor method)
NumpyAllocator (class in tinygrad.runtime.ops_cpu)
O
obj (in module tinygrad.nn.state)
,
[1]
offs (tinygrad.codegen.linearizer.Linearizer attribute)
offset (tinygrad.runtime.ops_disk.DiskBuffer attribute)
(tinygrad.shape.view.View attribute)
,
[1]
offsets (in module tinygrad.nn.state)
on_exit (tinygrad.helpers.Timing attribute)
ones() (tinygrad.tensor.Tensor static method)
ones_like() (tinygrad.tensor.Tensor method)
Op (in module tinygrad.ops)
,
[1]
op (tinygrad.codegen.kernel.Opt attribute)
,
[1]
(tinygrad.lazy.LazyBuffer attribute)
,
[1]
,
[2]
,
[3]
(tinygrad.ops.LazyOp attribute)
,
[1]
op_estimate (tinygrad.device.InterpretedASTRunner attribute)
OpenCLLanguage (class in tinygrad.renderer.cstyle)
OpenCLRenderer() (in module tinygrad.renderer.cstyle)
OpNode (class in tinygrad.shape.symbolic)
ops (tinygrad.lazy.LazyBuffer attribute)
Opt (class in tinygrad.codegen.kernel)
optimize_local_size() (in module tinygrad.features.search)
Optimizer (class in tinygrad.nn.optim)
optimizer (tinygrad.runtime.ops_llvm.LLVM attribute)
,
[1]
OptOps (class in tinygrad.codegen.kernel)
opts (tinygrad.codegen.kernel.Kernel attribute)
OpType (in module tinygrad.ops)
optype (tinygrad.lazy.LazyBuffer attribute)
out (tinygrad.ops.ScheduleItem attribute)
out_device (tinygrad.lazy.LazyBuffer attribute)
out_dtype (tinygrad.lazy.LazyBuffer attribute)
out_shape (tinygrad.lazy.LazyBuffer attribute)
output_buffer (tinygrad.lazy.LazyBuffer attribute)
output_dtype (tinygrad.renderer.cstyle.CStyleLanguage attribute)
output_padding (tinygrad.tensor.Tensor attribute)
output_shape (tinygrad.codegen.kernel.Kernel property)
output_type (in module tinygrad.renderer.llvmir)
output_type() (in module tinygrad.runtime.ops_cpu)
(in module tinygrad.runtime.ops_torch)
P
Pad (class in tinygrad.mlops)
PAD (tinygrad.ops.MovementOps attribute)
,
[1]
pad() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
(tinygrad.shape.shapetracker.ShapeTracker method)
(tinygrad.shape.view.View method)
(tinygrad.tensor.Tensor method)
pad2d() (tinygrad.tensor.Tensor method)
padding (in module tinygrad.features.image)
(tinygrad.tensor.Tensor attribute)
PADTO (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
params (in module tinygrad.nn.optim)
(tinygrad.nn.optim.LAMB attribute)
(tinygrad.nn.optim.Optimizer attribute)
(tinygrad.nn.optim.SGD attribute)
parents (tinygrad.tensor.Function attribute)
partition() (in module tinygrad.helpers)
Permute (class in tinygrad.mlops)
PERMUTE (tinygrad.ops.MovementOps attribute)
,
[1]
permute() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
(tinygrad.shape.shapetracker.ShapeTracker method)
(tinygrad.shape.view.View method)
(tinygrad.tensor.Tensor method)
PHI (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
PlaceHolder (class in tinygrad.jit)
pow() (tinygrad.tensor.Tensor method)
prefix (in module tinygrad.nn.state)
(tinygrad.helpers.Timing attribute)
prekernel (tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
pretty_ptx() (in module tinygrad.helpers)
prg (tinygrad.jit.JitItem attribute)
,
[1]
(tinygrad.runtime.ops_cuda.CUDAProgram attribute)
(tinygrad.runtime.ops_webgpu.WebGPUProgram attribute)
print_tree() (in module tinygrad.graph)
priority (tinygrad.helpers.DType attribute)
(tinygrad.helpers.ImageDType attribute)
prod() (in module tinygrad.helpers)
Profiling (class in tinygrad.helpers)
PtrDType (class in tinygrad.helpers)
Q
quick_gelu() (tinygrad.tensor.Tensor method)
R
r() (tinygrad.lazy.LazyBuffer method)
rand() (tinygrad.tensor.Tensor static method)
randint() (tinygrad.tensor.Tensor static method)
randn() (tinygrad.tensor.Tensor static method)
rawbufs (in module tinygrad.features.search)
,
[1]
,
[2]
(tinygrad.jit.JitItem attribute)
,
[1]
real_strides() (tinygrad.shape.shapetracker.ShapeTracker method)
realize() (tinygrad.nn.optim.Optimizer method)
(tinygrad.tensor.Tensor method)
realized (tinygrad.codegen.kernel.LocalBuffer attribute)
,
[1]
(tinygrad.lazy.LazyBuffer property)
(tinygrad.ops.LazyOp property)
RECIP (tinygrad.ops.UnaryOps attribute)
reciprocal() (tinygrad.tensor.Tensor method)
RedNode (class in tinygrad.shape.symbolic)
(tinygrad.shape.symbolic.AndNode attribute)
reduceop (tinygrad.codegen.kernel.Kernel attribute)
ReduceOps (class in tinygrad.ops)
Relu (class in tinygrad.mlops)
relu (tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
relu() (tinygrad.tensor.Tensor method)
relu6 (tinygrad.tensor.Tensor attribute)
relu6() (tinygrad.tensor.Tensor method)
render() (tinygrad.shape.symbolic.Node method)
render_cast() (tinygrad.renderer.cstyle.CStyleLanguage method)
(tinygrad.renderer.cstyle.WGSLLanguage method)
render_conditional() (tinygrad.renderer.cstyle.CStyleLanguage method)
(tinygrad.renderer.cstyle.WGSLLanguage method)
render_const() (tinygrad.renderer.cstyle.CStyleLanguage method)
(tinygrad.renderer.cstyle.WGSLLanguage method)
render_for() (tinygrad.renderer.cstyle.CStyleLanguage method)
(tinygrad.renderer.cstyle.WGSLLanguage method)
render_if() (tinygrad.renderer.cstyle.CStyleLanguage method)
(tinygrad.renderer.cstyle.WGSLLanguage method)
render_kernel() (tinygrad.renderer.cstyle.CStyleLanguage method)
(tinygrad.renderer.cstyle.WGSLLanguage method)
render_load() (tinygrad.renderer.cstyle.CStyleLanguage method)
render_local() (tinygrad.renderer.cstyle.CStyleLanguage method)
(tinygrad.renderer.cstyle.WGSLLanguage method)
render_ops (tinygrad.codegen.linearizer.Linearizer attribute)
render_store() (tinygrad.renderer.cstyle.CStyleLanguage method)
(tinygrad.renderer.cstyle.WGSLLanguage method)
repeat() (tinygrad.tensor.Tensor method)
repeats (tinygrad.tensor.Tensor attribute)
replace_with_movement_ops() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
requires_grad (tinygrad.tensor.Function attribute)
(tinygrad.tensor.Tensor attribute)
reset() (tinygrad.helpers.GlobalCounters static method)
(tinygrad.jit.TinyJit method)
Reshape (class in tinygrad.mlops)
RESHAPE (tinygrad.ops.MovementOps attribute)
,
[1]
reshape() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
(tinygrad.shape.shapetracker.ShapeTracker method)
(tinygrad.shape.view.View method)
(tinygrad.tensor.Tensor method)
reshape_and_permute() (tinygrad.codegen.kernel.Kernel method)
ret (tinygrad.jit.TinyJit attribute)
(tinygrad.lazy.LazyBuffer attribute)
(tinygrad.mlops.Max attribute)
(tinygrad.mlops.Relu attribute)
(tinygrad.mlops.Sigmoid attribute)
(tinygrad.mlops.Sqrt attribute)
ReturnType (class in tinygrad.jit)
round_up() (in module tinygrad.helpers)
rsqrt() (tinygrad.tensor.Tensor method)
run_schedule() (in module tinygrad.realize)
S
safe_load() (in module tinygrad.nn.state)
safe_load_metadata() (in module tinygrad.nn.state)
safe_save() (in module tinygrad.nn.state)
saved_exprs (tinygrad.codegen.linearizer.Linearizer.self attribute)
scalar() (tinygrad.helpers.DType method)
scaled_dot_product_attention() (tinygrad.tensor.Tensor method)
scaled_uniform() (tinygrad.tensor.Tensor static method)
schedule() (tinygrad.lazy.LazyBuffer method)
ScheduleItem (class in tinygrad.ops)
seed (tinygrad.tensor.Tensor attribute)
seen (tinygrad.lazy.LazyBuffer attribute)
self (in module tinygrad.features.image)
(tinygrad.codegen.linearizer.Linearizer attribute)
(tinygrad.lazy.LazyBuffer attribute)
,
[1]
,
[2]
,
[3]
,
[4]
,
[5]
,
[6]
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
,
[2]
,
[3]
,
[4]
,
[5]
,
[6]
(tinygrad.runtime.ops_gpu.CLDevice attribute)
(tinygrad.shape.shapetracker.ShapeTracker attribute)
,
[1]
,
[2]
(tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
,
[3]
,
[4]
,
[5]
,
[6]
,
[7]
,
[8]
,
[9]
,
[10]
,
[11]
,
[12]
,
[13]
,
[14]
,
[15]
,
[16]
,
[17]
,
[18]
sequential() (tinygrad.tensor.Tensor method)
set_kernel_node_launch_dims() (tinygrad.features.graph.cuda.CUDAGraph method)
SGD (class in tinygrad.nn.optim)
shape (tinygrad.helpers.ImageDType attribute)
(tinygrad.lazy.LazyBuffer attribute)
(tinygrad.ops.FlopCounter attribute)
,
[1]
(tinygrad.shape.shapetracker.ShapeTracker property)
(tinygrad.shape.view.View attribute)
,
[1]
(tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
,
[3]
,
[4]
,
[5]
,
[6]
(tinygrad.tensor.Tensor property)
shape_len (tinygrad.codegen.kernel.Kernel attribute)
(tinygrad.codegen.kernel.Kernel property)
(tinygrad.codegen.kernel.Kernel.self attribute)
shape_offsets() (tinygrad.codegen.kernel.Kernel method)
shape_to_axis() (in module tinygrad.runtime.ops_cpu)
ShapeTracker (class in tinygrad.shape.shapetracker)
shift_to() (tinygrad.codegen.kernel.Kernel method)
Shrink (class in tinygrad.mlops)
SHRINK (tinygrad.ops.MovementOps attribute)
,
[1]
shrink() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
(tinygrad.shape.shapetracker.ShapeTracker method)
(tinygrad.shape.view.View method)
(tinygrad.tensor.Tensor method)
Sigmoid (class in tinygrad.mlops)
sigmoid (tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
sigmoid() (tinygrad.tensor.Tensor method)
sign() (tinygrad.tensor.Tensor method)
silu() (tinygrad.tensor.Tensor method)
simplify (tinygrad.codegen.linearizer.Linearizer attribute)
simplify() (tinygrad.shape.shapetracker.ShapeTracker method)
simplify_merge_adjacent() (tinygrad.codegen.kernel.Kernel method)
simplify_ones() (tinygrad.codegen.kernel.Kernel method)
Sin (class in tinygrad.mlops)
SIN (tinygrad.ops.UnaryOps attribute)
sin() (tinygrad.tensor.Tensor method)
sinh() (tinygrad.tensor.Tensor method)
size (tinygrad.codegen.kernel.LocalBuffer attribute)
,
[1]
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.runtime.ops_disk.DiskBuffer attribute)
size() (tinygrad.shape.shapetracker.ShapeTracker method)
size_prefix (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
slice() (tinygrad.tensor.Tensor method)
smem_align (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
smem_prefix (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
smem_prefix_for_cast (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
softmax() (tinygrad.tensor.Tensor method)
softplus (tinygrad.tensor.Tensor attribute)
softplus() (tinygrad.tensor.Tensor method)
softsign() (tinygrad.tensor.Tensor method)
sort (tinygrad.helpers.Profiling attribute)
sparse_categorical_crossentropy() (tinygrad.tensor.Tensor method)
SPECIAL (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
Sqrt (class in tinygrad.mlops)
SQRT (tinygrad.ops.UnaryOps attribute)
sqrt (tinygrad.tensor.Tensor attribute)
,
[1]
sqrt() (tinygrad.tensor.Tensor method)
square (tinygrad.tensor.Tensor attribute)
,
[1]
square() (tinygrad.tensor.Tensor method)
squeeze() (tinygrad.tensor.Tensor method)
src (tinygrad.lazy.LazyBuffer attribute)
,
[1]
(tinygrad.ops.LazyOp attribute)
,
[1]
srcs (tinygrad.lazy.LazyBuffer attribute)
ss (tinygrad.tensor.Tensor attribute)
st (tinygrad.lazy.LazyBuffer attribute)
(tinygrad.ops.ConstBuffer attribute)
(tinygrad.ops.LazyOp property)
(tinygrad.ops.MemBuffer attribute)
stack (tinygrad.helpers.Context attribute)
stack() (tinygrad.tensor.Tensor static method)
start_dim (tinygrad.tensor.Tensor attribute)
state_dict (in module tinygrad.nn.state)
status (in module tinygrad.runtime.ops_cuda)
std (tinygrad.tensor.Tensor attribute)
std() (tinygrad.tensor.Tensor method)
step() (tinygrad.nn.optim.LAMB method)
(tinygrad.nn.optim.SGD method)
STORE (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
(tinygrad.ops.BufferOps attribute)
,
[1]
str_dtype() (in module tinygrad.graph)
strict (in module tinygrad.nn.state)
stride (in module tinygrad.features.image)
STRIDE (tinygrad.ops.MovementOps attribute)
,
[1]
stride (tinygrad.tensor.Tensor attribute)
,
[1]
,
[2]
stride() (tinygrad.lazy.LazyBuffer method)
(tinygrad.ops.LazyOp method)
(tinygrad.shape.shapetracker.ShapeTracker method)
(tinygrad.shape.view.View method)
strides (tinygrad.shape.view.View attribute)
,
[1]
strides_for_shape() (in module tinygrad.shape.view)
strip_parens() (in module tinygrad.helpers)
sts (tinygrad.codegen.kernel.Kernel attribute)
,
[1]
(tinygrad.codegen.kernel.Kernel.self attribute)
,
[1]
,
[2]
,
[3]
sts_backup (tinygrad.codegen.linearizer.Linearizer attribute)
Sub (class in tinygrad.mlops)
SUB (tinygrad.ops.BinaryOps attribute)
sub() (tinygrad.tensor.Tensor method)
substitute() (tinygrad.shape.symbolic.AndNode method)
(tinygrad.shape.symbolic.DivNode method)
(tinygrad.shape.symbolic.LtNode method)
(tinygrad.shape.symbolic.ModNode method)
(tinygrad.shape.symbolic.MulNode method)
(tinygrad.shape.symbolic.Node method)
(tinygrad.shape.symbolic.NumNode method)
(tinygrad.shape.symbolic.SumNode method)
(tinygrad.shape.symbolic.Variable method)
Sum (class in tinygrad.mlops)
SUM (tinygrad.ops.ReduceOps attribute)
sum() (tinygrad.shape.symbolic.Node static method)
(tinygrad.tensor.Tensor method)
SumNode (class in tinygrad.shape.symbolic)
super (tinygrad.runtime.ops_disk.DiskDevice attribute)
supports_float4 (tinygrad.codegen.kernel.LinearizerOptions attribute)
,
[1]
supports_float4_alu (tinygrad.codegen.kernel.LinearizerOptions attribute)
,
[1]
swish (tinygrad.tensor.Tensor attribute)
swish() (tinygrad.tensor.Tensor method)
sym_infer() (in module tinygrad.shape.symbolic)
sym_render() (in module tinygrad.shape.symbolic)
synchronize (tinygrad.device.Interpreted attribute)
synchronize() (tinygrad.device.Compiled method)
(tinygrad.runtime.ops_cuda.CUDADevice method)
,
[1]
(tinygrad.runtime.ops_gpu.CLDevice method)
sz (tinygrad.helpers.DType attribute)
T
t (in module tinygrad.nn.state)
T (tinygrad.tensor.Tensor property)
tan() (tinygrad.tensor.Tensor method)
tanh (tinygrad.tensor.Tensor attribute)
,
[1]
tanh() (tinygrad.tensor.Tensor method)
target_machine (tinygrad.runtime.ops_llvm.LLVM attribute)
,
[1]
temp() (in module tinygrad.helpers)
Tensor (class in tinygrad.tensor)
Tensor.train (class in tinygrad.tensor)
tensor_core (tinygrad.codegen.kernel.Kernel attribute)
tensor_type (in module tinygrad.nn.state)
TensorCore (class in tinygrad.codegen.kernel)
tensors (tinygrad.tensor.Function attribute)
(tinygrad.tensor.Tensor attribute)
TernaryOps (class in tinygrad.ops)
thread_local_aliases (tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
thread_local_sizes (tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
threads (tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
time_execution_cuda_style() (in module tinygrad.helpers)
time_linearizer() (in module tinygrad.features.search)
time_sum_s (tinygrad.helpers.GlobalCounters attribute)
Timing (class in tinygrad.helpers)
tinygrad.codegen.kernel
module
tinygrad.codegen.linearizer
module
tinygrad.device
module
tinygrad.features.graph.cuda
module
tinygrad.features.image
module
tinygrad.features.search
module
tinygrad.graph
module
tinygrad.helpers
module
tinygrad.jit
module
tinygrad.lazy
module
tinygrad.mlops
module
tinygrad.nn.optim
module
tinygrad.nn.state
module
tinygrad.ops
module
tinygrad.realize
module
tinygrad.renderer.cstyle
module
tinygrad.renderer.llvmir
module
tinygrad.runtime.ops_clang
module
tinygrad.runtime.ops_cpu
module
tinygrad.runtime.ops_cuda
module
tinygrad.runtime.ops_disk
module
tinygrad.runtime.ops_gpu
module
tinygrad.runtime.ops_llvm
module
tinygrad.runtime.ops_torch
module
tinygrad.runtime.ops_webgpu
module
tinygrad.shape.shapetracker
module
tinygrad.shape.symbolic
module
tinygrad.shape.view
module
tinygrad.tensor
module
TinyJit (class in tinygrad.jit)
to() (tinygrad.tensor.Tensor method)
to_() (tinygrad.tensor.Tensor method)
to_char_p_p() (in module tinygrad.helpers)
to_function_name() (in module tinygrad.helpers)
to_image_idx() (in module tinygrad.features.image)
to_movement_ops() (tinygrad.shape.shapetracker.ShapeTracker method)
to_program() (tinygrad.device.Compiled method)
to_tuple() (tinygrad.jit.PlaceHolder method)
toCPU() (tinygrad.device.Buffer method)
torch() (tinygrad.tensor.Tensor method)
torch_load() (in module tinygrad.nn.state)
TorchAllocator (class in tinygrad.runtime.ops_torch)
training (tinygrad.tensor.Tensor attribute)
,
[1]
transpose() (tinygrad.tensor.Tensor method)
tril() (tinygrad.tensor.Tensor method)
triu() (tinygrad.tensor.Tensor method)
trunc() (tinygrad.tensor.Tensor method)
tuplize_uops() (in module tinygrad.features.search)
type_map (tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
U
ud (tinygrad.runtime.ops_disk.DiskBuffer attribute)
uint16 (tinygrad.helpers.dtypes attribute)
uint32 (tinygrad.helpers.dtypes attribute)
uint64 (tinygrad.helpers.dtypes attribute)
uint8 (tinygrad.helpers.dtypes attribute)
UnaryOps (class in tinygrad.ops)
unbind() (tinygrad.shape.shapetracker.ShapeTracker method)
(tinygrad.shape.symbolic.Node method)
(tinygrad.shape.symbolic.Variable method)
(tinygrad.shape.view.View method)
UnderlyingDiskBuffer (class in tinygrad.runtime.ops_disk)
unif_samples (tinygrad.tensor.Tensor attribute)
uniform() (tinygrad.tensor.Tensor static method)
unit_stride_axes() (tinygrad.shape.shapetracker.ShapeTracker method)
UNROLL (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
unsqueeze() (tinygrad.tensor.Tensor method)
unwrap() (in module tinygrad.helpers)
unwrap2() (in module tinygrad.helpers)
UOp (class in tinygrad.codegen.linearizer)
uop (tinygrad.codegen.linearizer.Linearizer attribute)
(tinygrad.codegen.linearizer.UOp attribute)
,
[1]
uop() (tinygrad.codegen.linearizer.Linearizer method)
uop_alu_idx (tinygrad.codegen.linearizer.Linearizer attribute)
uop_alu_idx() (tinygrad.codegen.linearizer.Linearizer method)
UOps (class in tinygrad.codegen.linearizer)
uops (in module tinygrad.features.search)
(tinygrad.codegen.linearizer.Linearizer.self attribute)
uops_to_cstyle() (in module tinygrad.renderer.cstyle)
uops_to_llvm_ir() (in module tinygrad.renderer.llvmir)
upc_backup (tinygrad.codegen.linearizer.Linearizer attribute)
UPCAST (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
upcast() (tinygrad.codegen.kernel.Kernel method)
upcast_dim (tinygrad.codegen.kernel.TensorCore attribute)
,
[1]
upcast_in_mid_reduce_axes (tinygrad.codegen.kernel.Kernel attribute)
(tinygrad.codegen.kernel.Kernel property)
upcasted (tinygrad.codegen.kernel.Kernel attribute)
,
[1]
(tinygrad.codegen.kernel.Kernel.self attribute)
,
[1]
upcasted_axis() (tinygrad.codegen.kernel.Kernel method)
UPCASTMID (tinygrad.codegen.kernel.OptOps attribute)
,
[1]
update_stats() (in module tinygrad.device)
use_tensor_cores (tinygrad.codegen.kernel.Kernel attribute)
uses_ptr_arithmetic (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.MetalLanguage attribute)
uses_vload (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
V
val (in module tinygrad.renderer.llvmir)
(tinygrad.ops.ConstBuffer attribute)
(tinygrad.shape.symbolic.Variable property)
valid (in module tinygrad.features.image)
(in module tinygrad.shape.shapetracker)
value (tinygrad.helpers.ContextVar attribute)
(tinygrad.tensor.Tensor attribute)
var_dtype (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
,
[2]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
var_name (tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
var_vals (tinygrad.features.graph.cuda.CUDAGraph attribute)
(tinygrad.lazy.LazyBuffer attribute)
(tinygrad.ops.ScheduleItem attribute)
(tinygrad.shape.shapetracker.ShapeTracker property)
Variable (class in tinygrad.shape.symbolic)
vars() (tinygrad.shape.shapetracker.ShapeTracker method)
(tinygrad.shape.symbolic.Node method)
(tinygrad.shape.symbolic.OpNode method)
(tinygrad.shape.symbolic.RedNode method)
(tinygrad.shape.symbolic.Variable method)
(tinygrad.shape.view.View method)
vars_from_ast() (in module tinygrad.lazy)
vec() (tinygrad.helpers.DType method)
verbose (in module tinygrad.nn.state)
View (class in tinygrad.shape.view)
view (in module tinygrad.shape.shapetracker)
,
[1]
views (tinygrad.lazy.LazyBuffer attribute)
(tinygrad.shape.shapetracker.ShapeTracker attribute)
,
[1]
vin (tinygrad.codegen.linearizer.Linearizer attribute)
(tinygrad.codegen.linearizer.UOp attribute)
,
[1]
W
wd (in module tinygrad.nn.optim)
(tinygrad.nn.optim.LAMB attribute)
webgpu() (tinygrad.tensor.Tensor method)
WebGpuAllocator (class in tinygrad.runtime.ops_webgpu)
WebGpuDevice (class in tinygrad.runtime.ops_webgpu)
WebGPUProgram (class in tinygrad.runtime.ops_webgpu)
weight (in module tinygrad.features.image)
(tinygrad.tensor.Tensor attribute)
,
[1]
weight_decay (tinygrad.nn.optim.SGD attribute)
WGSLLanguage (class in tinygrad.renderer.cstyle)
WGSLRenderer() (in module tinygrad.renderer.cstyle)
Where (class in tinygrad.mlops)
WHERE (tinygrad.ops.TernaryOps attribute)
,
[1]
where() (tinygrad.tensor.Tensor method)
wino (tinygrad.tensor.Tensor attribute)
WMMA (tinygrad.codegen.linearizer.UOps attribute)
,
[1]
X
x (in module tinygrad.runtime.ops_torch)
,
[1]
,
[2]
(in module tinygrad.shape.symbolic)
(tinygrad.codegen.linearizer.Linearizer attribute)
(tinygrad.lazy.LazyBuffer attribute)
(tinygrad.mlops.Add attribute)
(tinygrad.mlops.Div attribute)
(tinygrad.mlops.Less attribute)
(tinygrad.mlops.Log attribute)
(tinygrad.mlops.Max attribute)
(tinygrad.mlops.Mul attribute)
(tinygrad.mlops.Relu attribute)
(tinygrad.mlops.Sin attribute)
(tinygrad.mlops.Sub attribute)
(tinygrad.mlops.Where attribute)
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
,
[2]
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
,
[1]
xid (tinygrad.renderer.cstyle.CStyleLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.CUDALanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.HIPLanguage attribute)
,
[1]
(tinygrad.renderer.cstyle.OpenCLLanguage attribute)
,
[1]
Y
y (in module tinygrad.runtime.ops_torch)
,
[1]
(tinygrad.mlops.Add attribute)
(tinygrad.mlops.Div attribute)
(tinygrad.mlops.Less attribute)
(tinygrad.mlops.Mul attribute)
(tinygrad.mlops.Sub attribute)
(tinygrad.renderer.cstyle.CStyleLanguage attribute)
(tinygrad.renderer.cstyle.WGSLLanguage attribute)
Z
Zero (class in tinygrad.mlops)
zero_grad() (tinygrad.nn.optim.Optimizer method)
zeros() (tinygrad.tensor.Tensor static method)
zeros_like() (tinygrad.tensor.Tensor method)