Struct Select 

pub struct Select {
    pub input: AnnotatedNode,
    pub buckets: AnnotatedNode,
}

Fields

input: AnnotatedNode

buckets: AnnotatedNode

Trait Implementations

impl Clone for Select

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Select

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<B: BackendMarker> GraphIROperationBase<B> for Select

fn nodes(&self) -> Vec<AnnotatedNode>

fn output_shape(&self, ir: &GraphIR<B>) -> Result<Shape, GraphIRError>

fn output_batched(&self, ir: &GraphIR<B>) -> Result<bool, GraphIRError>

fn output_requires_grad(&self, ir: &GraphIR<B>) -> Result<bool, GraphIRError>

fn output_layout( &self, _ir: &GraphIR<B>, ) -> Result<Option<NonZeroUsize>, GraphIRError>

fn ancillary_buffers( &self, _ir: &GraphIR<B>, ) -> Result<Vec<(Shape, Option<NonZeroUsize>, bool)>, GraphIRError>

fn shorthand(&self) -> String

impl<B: BackendMarker> GraphIROperationCompilable<B> for Select

where B::Backend: CoreDeviceOps,

fn forward_pass( &self, graph: &Graph<B::Backend>, output_node: NodeId, ) -> DeviceFunction<B::Backend>

fn backward_pass( &self, graph: &Graph<B::Backend>, output_node: NodeId, ) -> DeviceFunction<B::Backend>