Skip to main content

Module embedding

teeny_kernels::nn::loss

Module embedding

Structs§

CosineEmbeddingLossBackward: Cosine embedding loss backward (per-row).
CosineEmbeddingLossForward: Cosine embedding loss forward (per-row).
TripletMarginLossBackward: Triplet margin loss backward (per-row).
TripletMarginLossForward: Triplet margin loss forward (per-row).

Functions§

cosine_embedding_loss_backward: Cosine embedding loss backward (per-row).
cosine_embedding_loss_forward: Cosine embedding loss forward (per-row).
triplet_margin_loss_backward: Triplet margin loss backward (per-row).
triplet_margin_loss_forward: Triplet margin loss forward (per-row).