chenyuandGitHub c7368515d2 refactor sparse_categorical_crossentropy (#4406) ...

factor out the -1 * and / loss_mask.sum() for both smoothing and non-smoothing terms

2024-05-03 14:28:36 -04:00

..

codegen

NV tensor_cores in kernel.py (#4399)

2024-05-02 22:33:08 -04:00

engine

infra for scheduler process replay (#4405)

2024-05-03 20:29:13 +03:00

features

search: skip timing the unoptimized kernel (#4395)

2024-05-02 16:48:49 -04:00

nn

update docs (#4356)

2024-04-30 16:51:42 +09:00

renderer

fix define global (#4383)

2024-05-01 22:32:56 -04:00

runtime

factor out resource access logic in multigraph base class (#4385)

2024-05-03 00:38:22 +03:00

shape

add consecutive property to shapetracker

2024-05-02 10:41:28 -07:00

__init__.py

add mnist tutorial

2024-04-25 16:08:32 +08:00

buffer.py

new memory scheduler with explicit refcounts (#4198)

2024-04-17 08:46:47 +04:00

device.py

fix define global (#4383)

2024-05-01 22:32:56 -04:00

dtype.py

rename Scalar to ConstType and cast_scalar to as_const (#3946)

2024-03-26 22:39:58 -04:00

function.py

remove downcast_half in sum (#4376)

2024-05-01 11:46:44 -04:00

helpers.py

search: add additional 7 factors to the action space (#4256)

2024-04-22 19:14:23 -04:00

lazy.py

remove downcast_half in sum (#4376)

2024-05-01 11:46:44 -04:00

ops.py

tensor variable (#4362)

2024-04-30 14:08:57 -07:00

tensor.py

refactor sparse_categorical_crossentropy (#4406)

2024-05-03 14:28:36 -04:00