Module cpu

Expand description

CPU graph execution.

Structs§

batched_mat_mul
concatenate: Wrapper around ndarray::concatenate() that can handle an empty input list.
convolution
cpu_eval_graph
cpu_eval_graph_exec: Evaluate the given graph on the CPU, with the given batch size and inputs, returning the full execution state including profiling information.
cpu_flip
cpu_gather
cpu_slice
layernorm: Layernorm along the given axis of the tensor.
slice_info
softmax: Softmax along the given axis of the tensor. Implementation (and more importantly, the generic bounds) based on softmax within the onnxruntime crate