ruda-tensor-dispatch 0.2.6

Runtime backend dispatch for RUDA tensors.
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
use alloc::{boxed::Box, vec::Vec};
use core::{future::Future, ops::Range};
use ruda_tensor::{Backend, AutodiffContext, TensorPrimitive, TensorData, Scalar, Distribution, ExecutionError,
    FloatDType, IntDType, BoolDType, Shape, Slice};
use ruda_tensor::ops::*;
use ruda_tensor::tensor::{Device, FloatTensor, IntTensor, BoolTensor, QuantizedTensor, quantization::QuantizationParametersPrimitive};
use ruda_core::tensor::quantization::{QuantScheme, QuantPropagation};
use crate::{Dispatch, DispatchTensor, context::Selection};

mod tensor;
mod int_tensor;
mod bool_tensor;
mod qtensor;
mod activation;
mod module;
mod transaction;