Trait dfdx::tensor_ops::Device
source · pub trait Device<E: Dtype>: DeviceStorage + CopySlice<E> + TensorFromVec<E> + TensorFromVec<usize> + StackKernel<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> + VecMatKernel<E> + MatMatKernel<E> + VecVecKernel<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 + 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<GeLUKernelOp, E> + UnaryKernel<SigmoidKernelOp, E> + UnaryKernel<SinKernelOp, E> + UnaryKernel<SqrtKernelOp, E> + UnaryKernel<SquareKernelOp, E> + UnaryKernel<TanhKernelOp, E> + UnaryKernel<PowfKernelOp<E>, E> + UnaryKernel<PowiKernelOp, E> + BinaryKernel<BCEKernelOp, E> + BinaryKernel<HuberErrorKernelOp<E>, E> + BinaryKernel<MaximumKernelOp, E> + BinaryKernel<MinimumKernelOp, E> + AxpyKernel<E> { }Expand description
A DeviceStorage that requires all the tensor ops implementations