Module kn_graph::cpu

CPU graph execution.

Structs

• CalculatedValue
• ExecutionInfo

Functions

• batched_mat_mul
• convolution
• cpu_eval_graph
  Evaluate the given graph on the CPU, with the given batch size and inputs, returning the outputs.
• 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.
• 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

Type Aliases

• Tensor
  The core tensor type of this crate.