Skip to main content

Crate tl_cpu

Crate tl_cpu 

Source
Expand description

CPU テンソルバックエンド

Re-exports§

pub use tensor::CpuTensor;

Modules§

autograd
Autograd - 自動微分(CPU版)
backend_impl
GpuTensor トレイトの CpuTensor 実装 + 演算メソッド
device_impl
CpuDevice: IDevice トレイトの CPU 実装
ffi
CPU 版 FFI 関数 GPU 版 (tl_runtime/src/lib.rs) と同じ C シグネチャで CpuTensor を操作。 JIT の add_global_mapping で同じシンボル名にマッピングされる。
memory
tensor
CPU テンソル

Enums§

DType
テンソルのデータ型