LICENSE
MANIFEST.in
PYPI_README.md
README.md
compile_engine.bat
engine.cpp
engine.dll
export_onnx.py
nn_engine.py
pyproject.toml
setup.py
directcompute_nn/__init__.py
directcompute_nn/blur.hlsl
directcompute_nn/compile_engine.bat
directcompute_nn/engine.dll
directcompute_nn/matmul.hlsl
directcompute_nn/matmul_coarsened.hlsl
directcompute_nn/matmul_coarsened_2d.hlsl
directcompute_nn/matmul_tiled.hlsl
directcompute_nn/matmul_tiled_unroll.hlsl
directcompute_nn/nn_accumulate_scalar.hlsl
directcompute_nn/nn_adam.hlsl
directcompute_nn/nn_add.hlsl
directcompute_nn/nn_add_bias.hlsl
directcompute_nn/nn_argmax_correct.hlsl
directcompute_nn/nn_attention_backward_dk.hlsl
directcompute_nn/nn_attention_backward_dq.hlsl
directcompute_nn/nn_attention_backward_dv.hlsl
directcompute_nn/nn_attention_backward_full_dk.hlsl
directcompute_nn/nn_attention_backward_full_dq.hlsl
directcompute_nn/nn_attention_backward_full_dv.hlsl
directcompute_nn/nn_attention_forward.hlsl
directcompute_nn/nn_attention_forward_full_train.hlsl
directcompute_nn/nn_attention_forward_train.hlsl
directcompute_nn/nn_batchnorm_backward_dx.hlsl
directcompute_nn/nn_batchnorm_backward_reduce.hlsl
directcompute_nn/nn_batchnorm_forward.hlsl
directcompute_nn/nn_batchnorm_mean_var.hlsl
directcompute_nn/nn_bias_grad.hlsl
directcompute_nn/nn_bias_relu.hlsl
directcompute_nn/nn_causal_masked_softmax.hlsl
directcompute_nn/nn_col2im.hlsl
directcompute_nn/nn_concat_insert.hlsl
directcompute_nn/nn_conv_backprop_filters.hlsl
directcompute_nn/nn_conv_backprop_filters_tiled.hlsl
directcompute_nn/nn_conv_backprop_input.hlsl
directcompute_nn/nn_conv_backprop_input_fused.hlsl
directcompute_nn/nn_conv_forward.hlsl
directcompute_nn/nn_conv_forward_tiled.hlsl
directcompute_nn/nn_conv_grad_reshape.hlsl
directcompute_nn/nn_conv_grad_reshape_relu.hlsl
directcompute_nn/nn_conv_reduce_filters.hlsl
directcompute_nn/nn_conv_reshape.hlsl
directcompute_nn/nn_depthwise_conv_backward_filter.hlsl
directcompute_nn/nn_depthwise_conv_backward_input.hlsl
directcompute_nn/nn_depthwise_conv_forward.hlsl
directcompute_nn/nn_dice_loss.hlsl
directcompute_nn/nn_dice_loss_grad.hlsl
directcompute_nn/nn_dice_loss_sums.hlsl
directcompute_nn/nn_gelu.hlsl
directcompute_nn/nn_gelu_grad.hlsl
directcompute_nn/nn_global_avg_pool.hlsl
directcompute_nn/nn_global_avg_pool_backward.hlsl
directcompute_nn/nn_grad_accum.hlsl
directcompute_nn/nn_grad_norm_final.hlsl
directcompute_nn/nn_grad_sq_reduce.hlsl
directcompute_nn/nn_im2col.hlsl
directcompute_nn/nn_layernorm_backward_dx.hlsl
directcompute_nn/nn_layernorm_backward_param.hlsl
directcompute_nn/nn_layernorm_forward.hlsl
directcompute_nn/nn_leaky_relu.hlsl
directcompute_nn/nn_leaky_relu_grad.hlsl
directcompute_nn/nn_loss.hlsl
directcompute_nn/nn_matmul.hlsl
directcompute_nn/nn_matmul_coarsened.hlsl
directcompute_nn/nn_matmul_dgpu.hlsl
directcompute_nn/nn_matmul_heavy.hlsl
directcompute_nn/nn_matmul_medium.hlsl
directcompute_nn/nn_matmul_reduce.hlsl
directcompute_nn/nn_matmul_transpose_a.hlsl
directcompute_nn/nn_matmul_transpose_b.hlsl
directcompute_nn/nn_matmul_universal.hlsl
directcompute_nn/nn_maxpool_backward.hlsl
directcompute_nn/nn_maxpool_forward.hlsl
directcompute_nn/nn_relu.hlsl
directcompute_nn/nn_relu6.hlsl
directcompute_nn/nn_relu6_grad.hlsl
directcompute_nn/nn_relu_grad.hlsl
directcompute_nn/nn_rms.hlsl
directcompute_nn/nn_scale_add.hlsl
directcompute_nn/nn_sgd.hlsl
directcompute_nn/nn_sgd_clipped.hlsl
directcompute_nn/nn_sigmoid.hlsl
directcompute_nn/nn_sigmoid_grad.hlsl
directcompute_nn/nn_slice.hlsl
directcompute_nn/nn_softmax.hlsl
directcompute_nn/nn_softmax_ce_grad.hlsl
directcompute_nn/nn_upsample.hlsl
directcompute_nn/nn_upsample_backward.hlsl
directcompute_nn.egg-info/PKG-INFO
directcompute_nn.egg-info/SOURCES.txt
directcompute_nn.egg-info/dependency_links.txt
directcompute_nn.egg-info/requires.txt
directcompute_nn.egg-info/top_level.txt