Convolution 1D optimized axis map (#8628)

Summary:

Resubmit D67692963 (#8312)
Add the kOptimizedAxisMap flag for conv1D.

Reviewed By: jorgep31415

Differential Revision: D70009298

Author: nathanaelsee

backends/vulkan/runtime/graph/ops/impl/Convolution.cpp

@@ -475,7 +475,12 @@ void add_conv1d_node(
     const ValueRef out,
     const bool clamp_out) {
   ValueRef arg_weight = prepack_standard(
-      graph, weight, graph.storage_type_of(out), utils::kChannelsPacked);
+      graph,
+      weight,
+      graph.storage_type_of(out),
+      utils::kChannelsPacked,
+      /* passthrough = */ false,
+      utils::kOptimizedAxisMap);
   ValueRef arg_bias = prepack_biases(
       graph,
       bias,