Module kn_graph::cpu

CPU graph execution.

Structs

• CalculatedValue
• ExecutionInfo

Functions

• 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