Fix ArgMin op

Fix regression introduced in fee3f09 where ArgMin dispatched to `arg_max` instead of `arg_min`.
robertknight · Feb 10, 2025 · a4d9768 · a4d9768
1 parent 83f0003
commit a4d9768
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ops/reduce.rs b/src/ops/reduce.rs
@@ -168,7 +168,7 @@ impl Operator for ArgMin {
 
     fn run(&self, pool: &TensorPool, inputs: InputList) -> Result<OutputList, OpError> {
         let input = inputs.require(0)?;
-        dispatch_single_axis_reduce_op!(pool, input, arg_max, self.axis, self.keep_dims)
+        dispatch_single_axis_reduce_op!(pool, input, arg_min, self.axis, self.keep_dims)
     }
 }