Trait moose::kernels::PlacementSum

source · [−]

pub trait PlacementSum<S: Session, T, O> {
    fn sum(&self, sess: &S, axis: Option<usize>, x: &T) -> O;
}

Expand description

Sum along axis

Required Methods

fn sum(&self, sess: &S, axis: Option<usize>, x: &T) -> O

Implementors

impl PlacementSum<AsyncSession, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>> for HostPlacement

impl PlacementSum<AsyncSession, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>> for HostPlacement

impl PlacementSum<AsyncSession, AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>>, AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>>> for HostPlacement

impl PlacementSum<AsyncSession, AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>>, AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>>> for ReplicatedPlacement

impl PlacementSum<AsyncSession, HostFixedTensor<HostRingTensor<u64>>, HostFixedTensor<HostRingTensor<u64>>> for HostPlacement

impl PlacementSum<AsyncSession, HostFixedTensor<HostRingTensor<u128>>, HostFixedTensor<HostRingTensor<u128>>> for HostPlacement

impl PlacementSum<AsyncSession, HostRingTensor<u64>, HostRingTensor<u64>> for HostPlacement

impl PlacementSum<AsyncSession, HostRingTensor<u128>, HostRingTensor<u128>> for HostPlacement

impl PlacementSum<AsyncSession, HostTensor<f32>, HostTensor<f32>> for HostPlacement

impl PlacementSum<AsyncSession, HostTensor<f64>, HostTensor<f64>> for HostPlacement

impl PlacementSum<AsyncSession, RepFixedTensor<RepTensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>> for ReplicatedPlacement

impl PlacementSum<AsyncSession, RepFixedTensor<RepTensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>> for ReplicatedPlacement

impl PlacementSum<AsyncSession, RepTensor<HostRingTensor<u64>>, RepTensor<HostRingTensor<u64>>> for ReplicatedPlacement

impl PlacementSum<AsyncSession, RepTensor<HostRingTensor<u128>>, RepTensor<HostRingTensor<u128>>> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>> as PartiallySymbolicType>::Type, <FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>> as PartiallySymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>> as SymbolicType>::Type, <FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>> as PartiallySymbolicType>::Type, <FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>> as PartiallySymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>> as SymbolicType>::Type, <FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> as PartiallySymbolicType>::Type, <AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> as PartiallySymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> as PartiallySymbolicType>::Type, <AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> as PartiallySymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> as SymbolicType>::Type, <AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> as SymbolicType>::Type, <AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> as SymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <HostFixedTensor<HostRingTensor<u64>> as PartiallySymbolicType>::Type, <HostFixedTensor<HostRingTensor<u64>> as PartiallySymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <HostFixedTensor<HostRingTensor<u64>> as SymbolicType>::Type, <HostFixedTensor<HostRingTensor<u64>> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <HostFixedTensor<HostRingTensor<u128>> as PartiallySymbolicType>::Type, <HostFixedTensor<HostRingTensor<u128>> as PartiallySymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <HostFixedTensor<HostRingTensor<u128>> as SymbolicType>::Type, <HostFixedTensor<HostRingTensor<u128>> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <HostRingTensor<u64> as SymbolicType>::Type, <HostRingTensor<u64> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <HostRingTensor<u128> as SymbolicType>::Type, <HostRingTensor<u128> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <HostTensor<f32> as SymbolicType>::Type, <HostTensor<f32> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <HostTensor<f64> as SymbolicType>::Type, <HostTensor<f64> as SymbolicType>::Type> for HostPlacement

impl PlacementSum<SymbolicSession, <RepFixedTensor<RepTensor<HostRingTensor<u64>>> as PartiallySymbolicType>::Type, <RepFixedTensor<RepTensor<HostRingTensor<u64>>> as PartiallySymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <RepFixedTensor<RepTensor<HostRingTensor<u64>>> as SymbolicType>::Type, <RepFixedTensor<RepTensor<HostRingTensor<u64>>> as SymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <RepFixedTensor<RepTensor<HostRingTensor<u128>>> as PartiallySymbolicType>::Type, <RepFixedTensor<RepTensor<HostRingTensor<u128>>> as PartiallySymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <RepFixedTensor<RepTensor<HostRingTensor<u128>>> as SymbolicType>::Type, <RepFixedTensor<RepTensor<HostRingTensor<u128>>> as SymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <RepTensor<HostRingTensor<u64>> as PartiallySymbolicType>::Type, <RepTensor<HostRingTensor<u64>> as PartiallySymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <RepTensor<HostRingTensor<u64>> as SymbolicType>::Type, <RepTensor<HostRingTensor<u64>> as SymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <RepTensor<HostRingTensor<u128>> as PartiallySymbolicType>::Type, <RepTensor<HostRingTensor<u128>> as PartiallySymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SymbolicSession, <RepTensor<HostRingTensor<u128>> as SymbolicType>::Type, <RepTensor<HostRingTensor<u128>> as SymbolicType>::Type> for ReplicatedPlacement

impl PlacementSum<SyncSession, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>> for HostPlacement

impl PlacementSum<SyncSession, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>> for HostPlacement

impl PlacementSum<SyncSession, AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>>, AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>>> for HostPlacement

impl PlacementSum<SyncSession, AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>>, AbstractTensor<FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, FloatTensor<HostTensor<f32>, Mir3Tensor<HostTensor<f32>>>, FloatTensor<HostTensor<f64>, Mir3Tensor<HostTensor<f64>>>, BoolTensor<HostBitTensor, RepTensor<HostBitTensor>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>>> for ReplicatedPlacement

impl PlacementSum<SyncSession, HostFixedTensor<HostRingTensor<u64>>, HostFixedTensor<HostRingTensor<u64>>> for HostPlacement

impl PlacementSum<SyncSession, HostFixedTensor<HostRingTensor<u128>>, HostFixedTensor<HostRingTensor<u128>>> for HostPlacement

impl PlacementSum<SyncSession, HostRingTensor<u64>, HostRingTensor<u64>> for HostPlacement

impl PlacementSum<SyncSession, HostRingTensor<u128>, HostRingTensor<u128>> for HostPlacement

impl PlacementSum<SyncSession, HostTensor<f32>, HostTensor<f32>> for HostPlacement

impl PlacementSum<SyncSession, HostTensor<f64>, HostTensor<f64>> for HostPlacement

impl PlacementSum<SyncSession, RepFixedTensor<RepTensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>> for ReplicatedPlacement

impl PlacementSum<SyncSession, RepFixedTensor<RepTensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>> for ReplicatedPlacement

impl PlacementSum<SyncSession, RepTensor<HostRingTensor<u64>>, RepTensor<HostRingTensor<u64>>> for ReplicatedPlacement

impl PlacementSum<SyncSession, RepTensor<HostRingTensor<u128>>, RepTensor<HostRingTensor<u128>>> for ReplicatedPlacement