Skip to content

Commit cae89c5

Browse files
authored
Convolution 1D optimized axis map
Differential Revision: D70009298 Pull Request resolved: #8628
1 parent 79580a7 commit cae89c5

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

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

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -475,7 +475,12 @@ void add_conv1d_node(
475475
const ValueRef out,
476476
const bool clamp_out) {
477477
ValueRef arg_weight = prepack_standard(
478-
graph, weight, graph.storage_type_of(out), utils::kChannelsPacked);
478+
graph,
479+
weight,
480+
graph.storage_type_of(out),
481+
utils::kChannelsPacked,
482+
/* passthrough = */ false,
483+
utils::kOptimizedAxisMap);
479484
ValueRef arg_bias = prepack_biases(
480485
graph,
481486
bias,

0 commit comments

Comments
 (0)