File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
backends/vulkan/runtime/graph/ops/impl Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -42,11 +42,11 @@ void resize_embedding_node(
4242 const std::vector<ValueRef>& resize_args) {
4343 (void )resize_args;
4444 const ValueRef out = args.at (0 ).refs .at (0 );
45- const ValueRef weight = args.at (1 ).refs .at (0 );
46- const ValueRef indices = args.at (1 ).refs .at (1 );
45+ const ValueRef indices = args.at (1 ).refs .at (0 );
46+ const ValueRef weight = args.at (1 ).refs .at (1 );
4747
48- const std::vector<int64_t > weight_sizes = graph->sizes_of (weight);
4948 const std::vector<int64_t > indices_sizes = graph->sizes_of (indices);
49+ const std::vector<int64_t > weight_sizes = graph->sizes_of (weight);
5050
5151 // Output shape is indices.shape + [embedding_dim]
5252 // where embedding_dim is the last dimension of weight
You can’t perform that action at this time.
0 commit comments