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 で同じシンボル名にマッピングされる。
fused_ops
融合カーネル — CPU 実装
graph
CpuGraph — CPU バックエンドのキャプチャ・リプレイ
memory
stream
CpuStream — CPU バックエンドの GpuStream 実装
tensor
CPU テンソル

Enums§

DType
テンソルのデータ型