Ivar Flakstad 8124d1003f Affine metal kernel works. Need to extract buffer contents based on layout offset (like CudaSlice.slice) for candle intergration

2023-11-06 04:46:56 +01:00

..

activation.rs

Add support for the marian base model. (#1221)

2023-10-30 19:20:36 +00:00

batch_norm.rs

Add fuse-conv-bn method for Conv2d (#1196)

2023-10-27 15:56:50 +01:00

conv.rs

Add fuse-conv-bn method for Conv2d (#1196)

2023-10-27 15:56:50 +01:00

embedding.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

func.rs

Allow for different behavior between training and eval (#1213)

2023-10-29 07:53:09 +01:00

group_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

init.rs

VarBuilder cleanup (#627)

2023-08-27 18:03:26 +01:00

layer_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

lib.rs

Allow for different behavior between training and eval (#1213)

2023-10-29 07:53:09 +01:00

linear.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

loss.rs

Add Binary Cross Entropy With Logit Loss to nn crate (#1157)

2023-10-23 17:12:44 +01:00

ops.rs

Affine metal kernel works. Need to extract buffer contents based on layout offset (like CudaSlice.slice) for candle intergration

2023-11-06 04:46:56 +01:00

optim.rs

Only optimize float tensors. (#1069)

2023-10-10 09:05:41 +01:00

rnn.rs

More general seq forward functions for RNNs. (#1050)

2023-10-07 15:08:01 +01:00

sequential.rs

Make func cloneable. (#1137)

2023-10-20 16:28:50 +01:00

var_builder.rs

feat: add pth varbuilder (#1108)

2023-10-16 16:20:36 +01:00

var_map.rs

Use AsRef<str> for set_one. (#1033)

2023-10-05 06:05:44 +01:00