MCPcopy
hub / github.com/tinygrad/tinygrad / test_softmax

Function test_softmax

extra/nv_pma/collect.py:69–71  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

67
68@pcsampling_test("test_softmax")
69def test_softmax():
70 a = Tensor(np.random.rand(64, 64).astype(np.float32))
71 a.softmax().realize()
72
73@pcsampling_test("test_conv2d")
74def test_conv2d():

Callers

nothing calls this directly

Calls 4

TensorClass · 0.90
randMethod · 0.80
realizeMethod · 0.80
softmaxMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…