Docs.rs
  • rai-core-0.11.0
    • rai-core 0.11.0
    • Docs.rs crate page
    • MIT OR Apache-2.0
    • Links
    • Homepage
    • Repository
    • crates.io
    • Source
    • Owners
    • cksac
    • Dependencies
      • candle-core ^0.4.1 normal optional
      • candle-flash-attn ^0.4.1 normal optional
      • colored ^2.1.0 normal
      • dyn-clone ^1.0.16 normal
      • half ^2.3.1 normal
      • num-traits ^0.2.17 normal
      • once_cell ^1.19.0 normal
      • paste ^1.0 normal
      • safetensors ^0.4.1 normal
      • thiserror ^1.0.51 normal
      • tracing ^0.1.40 normal
    • Versions
    • 4.36% of the crate is documented
  • Platform
    • i686-pc-windows-msvc
    • i686-unknown-linux-gnu
    • x86_64-apple-darwin
    • x86_64-pc-windows-msvc
    • x86_64-unknown-linux-gnu
  • Feature flags
  • docs.rs
    • About docs.rs
    • Badges
    • Builds
    • Metadata
    • Shorthand URLs
    • Download
    • Rustdoc JSON
    • Build queue
    • Privacy policy
  • Rust
    • Rust website
    • The Book
    • Standard Library API Reference
    • Rust by Example
    • The Cargo Guide
    • Clippy Documentation

rai_core0.11.0

Module primitives

Module Items

  • Structs
  • Traits

In crate rai_core

rai_core

Module primitives

Source

Structs§

Abs
Add
Arange
ArgMax
ArgMin
AvgPool1d
AvgPool2d
Broadcast
Concatenate
Conv1d
Conv2d
ConvTranspose1d
ConvTranspose2d
Cos
Div
Equal
Erf
Exp
FlashAttention
FromArray
Full
Gather
Greater
GreaterEqual
IndexAdd
IndexSelect
Less
LessEqual
Log
Log2
Log10
LogSoftmax
MatMul
MaxPool1d
MaxPool2d
Maximum
Minimum
Mul
Narrow
Negative
Normal
NotEqual
Permute
PowerFloat
Random
ReduceMax
ReduceMin
ReduceSum
Reshape
Rsqrt
ScatterAdd
Sign
Sin
Softmax
Sqrt
Square
Sub
Tanh
ToContiguous
ToDType
ToDevice
Transpose
UpsampleNearest1d
UpsampleNearest2d
Where

Traits§

Primitive