From 223857251d7557310fa2adf3c11b0b8dfbdd8b68 Mon Sep 17 00:00:00 2001
From: Foke Singh <foke.singh@deepkapha.com>
Date: Tue, 26 Dec 2023 10:45:52 +0000
Subject: [PATCH] Update MNIST example to use new operator source locations and
 reflect recent API changes (sdpa bias removal, gatherSlices shape literals,
 Shape.outer, Tensor.all, and argMax dtype inference)

---
 examples/mnist/mnist.zig | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/mnist/mnist.zig b/examples/mnist/mnist.zig
index 0f35c7e..51ca1e4 100644
--- a/examples/mnist/mnist.zig
+++ b/examples/mnist/mnist.zig
@@ -33,7 +33,7 @@ const Mnist = struct {
         for (layers) |layer| {
             x = zml.call(layer, .forward, .{x});
         }
-        return x.argMax(0, .u8).indices;
+        return x.argMax(0).indices.convert(.u8);
     }
 };