Trait moose::kernels::PlacementArgmax

source · [−]

pub trait PlacementArgmax<S: Session, T, O> {
    fn argmax(&self, sess: &S, axis: usize, upmost_index: usize, x: &T) -> O;
}

Required Methods

fn argmax(&self, sess: &S, axis: usize, upmost_index: usize, x: &T) -> O

Implementors

impl PlacementArgmax<AsyncSession, FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> for HostPlacement

impl PlacementArgmax<AsyncSession, FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> for ReplicatedPlacement

impl PlacementArgmax<AsyncSession, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> for HostPlacement

impl PlacementArgmax<AsyncSession, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> for ReplicatedPlacement

impl PlacementArgmax<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 PlacementArgmax<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 PlacementArgmax<AsyncSession, HostFixedTensor<HostRingTensor<u64>>, HostTensor<u64>> for HostPlacement

impl PlacementArgmax<AsyncSession, HostFixedTensor<HostRingTensor<u128>>, HostTensor<u64>> for HostPlacement

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

impl PlacementArgmax<AsyncSession, HostRingTensor<u128>, HostRingTensor<u64>> for HostPlacement

impl PlacementArgmax<AsyncSession, RepFixedTensor<RepTensor<HostRingTensor<u64>>>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> for ReplicatedPlacement

impl PlacementArgmax<AsyncSession, RepFixedTensor<RepTensor<HostRingTensor<u128>>>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> for ReplicatedPlacement

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

impl PlacementArgmax<AsyncSession, RepTensor<HostRingTensor<u128>>, RepTensor<HostRingTensor<u64>>> for ReplicatedPlacement

impl PlacementArgmax<SymbolicSession, <FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>> as PartiallySymbolicType>::Type, <AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> as PartiallySymbolicType>::Type> for HostPlacement

impl PlacementArgmax<SymbolicSession, <FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>> as PartiallySymbolicType>::Type, <AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> as PartiallySymbolicType>::Type> for ReplicatedPlacement

impl PlacementArgmax<SymbolicSession, <FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>> as SymbolicType>::Type, <AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> as SymbolicType>::Type> for HostPlacement

impl PlacementArgmax<SymbolicSession, <FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>> as SymbolicType>::Type, <AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> as SymbolicType>::Type> for ReplicatedPlacement

impl PlacementArgmax<SymbolicSession, <FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>> as PartiallySymbolicType>::Type, <AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> as PartiallySymbolicType>::Type> for HostPlacement

impl PlacementArgmax<SymbolicSession, <FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>> as PartiallySymbolicType>::Type, <AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> as PartiallySymbolicType>::Type> for ReplicatedPlacement

impl PlacementArgmax<SymbolicSession, <FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>> as SymbolicType>::Type, <AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> as SymbolicType>::Type> for HostPlacement

impl PlacementArgmax<SymbolicSession, <FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>> as SymbolicType>::Type, <AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> as SymbolicType>::Type> for ReplicatedPlacement

impl PlacementArgmax<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 PlacementArgmax<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 PlacementArgmax<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 PlacementArgmax<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 PlacementArgmax<SymbolicSession, <HostFixedTensor<HostRingTensor<u64>> as SymbolicType>::Type, <HostTensor<u64> as SymbolicType>::Type> for HostPlacement

impl PlacementArgmax<SymbolicSession, <HostFixedTensor<HostRingTensor<u128>> as SymbolicType>::Type, <HostTensor<u64> as SymbolicType>::Type> for HostPlacement

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

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

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

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

impl PlacementArgmax<SymbolicSession, <RepFixedTensor<RepTensor<HostRingTensor<u128>>> as PartiallySymbolicType>::Type, <RepUintTensor<RepTensor<HostRingTensor<u64>>> as PartiallySymbolicType>::Type> for ReplicatedPlacement

impl PlacementArgmax<SymbolicSession, <RepFixedTensor<RepTensor<HostRingTensor<u128>>> as SymbolicType>::Type, <RepUintTensor<RepTensor<HostRingTensor<u64>>> as SymbolicType>::Type> for ReplicatedPlacement

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

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

impl PlacementArgmax<SyncSession, FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> for HostPlacement

impl PlacementArgmax<SyncSession, FixedTensor<HostFixedTensor<HostRingTensor<u64>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u64>>>, RepFixedTensor<RepTensor<HostRingTensor<u64>>>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> for ReplicatedPlacement

impl PlacementArgmax<SyncSession, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> for HostPlacement

impl PlacementArgmax<SyncSession, FixedTensor<HostFixedTensor<HostRingTensor<u128>>, MirFixedTensor<Mir3Tensor<HostRingTensor<u128>>>, RepFixedTensor<RepTensor<HostRingTensor<u128>>>>, AbstractUint64Tensor<HostTensor<u64>, RepUintTensor<RepTensor<HostRingTensor<u64>>>>> for ReplicatedPlacement

impl PlacementArgmax<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 PlacementArgmax<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 PlacementArgmax<SyncSession, HostFixedTensor<HostRingTensor<u64>>, HostTensor<u64>> for HostPlacement

impl PlacementArgmax<SyncSession, HostFixedTensor<HostRingTensor<u128>>, HostTensor<u64>> for HostPlacement

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

impl PlacementArgmax<SyncSession, HostRingTensor<u128>, HostRingTensor<u64>> for HostPlacement

impl PlacementArgmax<SyncSession, RepFixedTensor<RepTensor<HostRingTensor<u64>>>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> for ReplicatedPlacement

impl PlacementArgmax<SyncSession, RepFixedTensor<RepTensor<HostRingTensor<u128>>>, RepUintTensor<RepTensor<HostRingTensor<u64>>>> for ReplicatedPlacement

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

impl PlacementArgmax<SyncSession, RepTensor<HostRingTensor<u128>>, RepTensor<HostRingTensor<u64>>> for ReplicatedPlacement