Struct CustomOpDefinition 

pub struct CustomOpDefinition {
    pub name: String,
    pub op_type: CustomOpType,
    pub input_shapes: Vec<Option<Vec<usize>>>,
    pub output_shapes: Vec<Option<Vec<usize>>>,
    pub cpu_source: Option<String>,
    pub cuda_source: Option<String>,
    pub compile_flags: Vec<String>,
    pub schema: OpSchema,
}

Custom operation definition

Fields

name: String

Operation name

op_type: CustomOpType

Operation type (forward, backward, both)

input_shapes: Vec<Option<Vec<usize>>>

Input tensor shapes (None means dynamic)

output_shapes: Vec<Option<Vec<usize>>>

Output tensor shapes (None means dynamic)

cpu_source: Option<String>

CPU implementation source

cuda_source: Option<String>

CUDA implementation source

compile_flags: Vec<String>

Custom compile flags for this operation

schema: OpSchema

Operation schema for validation

Trait Implementations

impl Clone for CustomOpDefinition

fn clone(&self) -> CustomOpDefinition

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CustomOpDefinition

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

Formats the value using the given formatter.