MCPcopy
hub / github.com/tinygrad/tinygrad / gradient.py

File gradient.py

tinygrad/gradient.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1from typing import cast
2import math, dataclasses, itertools
3from tinygrad.uop.ops import UOp, PatternMatcher, UPat, Ops, all_metadata, graph_rewrite
4from tinygrad.helpers import argsort

Callers

nothing calls this directly

Calls 15

PatternMatcherClass · 0.90
UPatClass · 0.90
sum_acc_dtypeFunction · 0.90
argsortFunction · 0.90
UOpClass · 0.90
reduce_gradientFunction · 0.85
sinMethod · 0.80
logMethod · 0.80
eqMethod · 0.80
powMethod · 0.80
log2Method · 0.80
reshapeMethod · 0.80

Tested by

no test coverage detected