burn_tensor/tensor/ops/
alias.rs

1use crate::backend::Backend;
2
3// We provide some type aliases to improve the readability of using associated types without
4// having to use the disambiguation syntax.
5
6/// Device type used by the backend.
7pub type Device<B> = <B as Backend>::Device;
8
9/// Float element type used by backend.
10pub type FloatElem<B> = <B as Backend>::FloatElem;
11/// Integer element type used by backend.
12pub type IntElem<B> = <B as Backend>::IntElem;
13
14/// Float tensor primitive type used by the backend.
15pub type FloatTensor<B> = <B as Backend>::FloatTensorPrimitive;
16/// Integer tensor primitive type used by the backend.
17pub type IntTensor<B> = <B as Backend>::IntTensorPrimitive;
18/// Boolean tensor primitive type used by the backend.
19pub type BoolTensor<B> = <B as Backend>::BoolTensorPrimitive;
20/// Quantized tensor primitive type used by the backend.
21pub type QuantizedTensor<B> = <B as Backend>::QuantizedTensorPrimitive;