Compiler.Xla.HLO.Builder.Lib.Arithmetic

For internal spidr use only.

Definitions

argMax : (HasIO io, Primitive outputType) => XlaOp -> Nat -> io XlaOp

Totality: total
Visibility: export