Trait dfdx::tensor_ops::Device
source · pub trait Device<E: Dtype>: Storage<E> + RandomU64 + CopySlice<E> + TensorFromVec<E> + TensorFromVec<usize> + TriangleTensor<E> + StackKernel<E> + ConcatKernel<E> + ConcatAlongKernel<E> + AdamKernel<E> + SgdKernel<E> + RMSpropKernel<E> + ZerosTensor<E> + OnesTensor<E> + SampleTensor<E> + OneFillStorage<E> + ZeroFillStorage<E> + SumKernel<E> + MaxReduceKernel<E> + MinReduceKernel<E> + ReshapeKernel<E> + ReplaceDimKernel<E> + RemoveDimKernel<E> + ChooseKernel<E> + SliceKernel<E> + RollKernel<E> + MatMatKernel<E> + MatMatBrKernel<E> + MatMatBatch3Kernel<E> + MatMatBatch4Kernel<E> + UnaryKernel<ScalarAddKernelOp<E>, E> + UnaryKernel<ScalarSubKernelOp<E>, E> + UnaryKernel<ScalarMulKernelOp<E>, E> + UnaryKernel<ScalarDivKernelOp<E>, E> + BinaryKernel<BinaryAddKernelOp, E> + BinaryKernel<BinarySubKernelOp, E> + BinaryKernel<BinaryMulKernelOp, E> + BinaryKernel<BinaryDivKernelOp, E> + BooleanKernel + CmpKernel<EqKernelOp, E> + CmpKernel<NeKernelOp, E> + CmpKernel<GtKernelOp, E> + CmpKernel<GeKernelOp, E> + CmpKernel<LtKernelOp, E> + CmpKernel<LeKernelOp, E> + ScalarCmpKernel<EqKernelOp, E> + ScalarCmpKernel<NeKernelOp, E> + ScalarCmpKernel<GtKernelOp, E> + ScalarCmpKernel<GeKernelOp, E> + ScalarCmpKernel<LtKernelOp, E> + ScalarCmpKernel<LeKernelOp, E> + UnaryKernel<AbsKernelOp, E> + UnaryKernel<ClampKernelOp<E>, E> + UnaryKernel<CosKernelOp, E> + DropoutKernel<E> + UnaryKernel<ExpKernelOp, E> + UnaryKernel<LnKernelOp, E> + UnaryKernel<NansToKernelOp<E>, E> + UnaryKernel<NegateKernelOp, E> + UnaryKernel<ReLUKernelOp, E> + UnaryKernel<FastGeLUKernelOp, E> + UnaryKernel<AccurateGeLUKernelOp, E> + UnaryKernel<SigmoidKernelOp, E> + UnaryKernel<SinKernelOp, E> + UnaryKernel<SqrtKernelOp, E> + UnaryKernel<SquareKernelOp, E> + UnaryKernel<TanhKernelOp, E> + UnaryKernel<PowfKernelOp<E>, E> + UnaryKernel<PowiKernelOp, E> + UnaryKernel<RecipKernelOp, E> + ToDtypeKernel<f32, E> + ToDtypeKernel<f64, E> + ToDtypeKernel<E, f32> + ToDtypeKernel<E, f64> + BinaryKernel<BCEKernelOp, E> + BinaryKernel<HuberErrorKernelOp<E>, E> + BinaryKernel<MaximumKernelOp, E> + BinaryKernel<MinimumKernelOp, E> + AxpyKernel<E> { }Expand description
A Storage that requires all the tensor ops implementations