operations.md

Operations

Check indictates support

• Pointwise Single

  • abs

  • neg

  • average_pool

  • batch_norm

  • relu

  • leakyrelu

  • sigmoid

• Pointwise Scalar (broadcast?)

  • add_scalar
  • sub_scalar_lh (s - t)
  • sub_scalar_rh (t - s)
  • mul_scalar
  • div_scalar_lh (s / t)
  • div_scalar_rh (t / s)

• Pointwise Double

  • add
  • sub
  • mul
  • div

• Reduction

  • max
  • min
  • sum
  • product

• Creation

  • zeroes
  • ones

• Utility

  • shape
  • stride
  • iter
  • slice
  • reshape
  • concat

• Not Supported (Out of scope operations or units that aren't supported)

  • affine_grid
  • and
  • arg_max
  • arg_min