Skip to content

Commit

Permalink
#0: snap to tile fix
Browse files Browse the repository at this point in the history
  • Loading branch information
mywoodstock committed May 17, 2024
1 parent e29dcca commit 996b29d
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -1338,7 +1338,7 @@ operation::ProgramWithCallbacks multi_core_optimized_conv_sharded_v2_new(const T
uint32_t pad_h = (uint32_t) conv_params[4];
uint32_t pad_w = (uint32_t) conv_params[5];
SlidingWindowConfig sliding_window_config = SlidingWindowConfig(input_tensor_shape[0], input_tensor_shape[1], input_tensor_shape[2], weight_size_h, weight_size_w, stride_h, stride_w,
pad_h, pad_w, 1, 1, parallelization_config.num_cores_nhw, parallel_config.grid);
pad_h, pad_w, 1, 1, parallelization_config.num_cores_nhw, parallel_config.grid, true);

// create conv config tensors
auto pad_metadata = sliding_window::generate_pad_metadata(sliding_window_config);
Expand Down

0 comments on commit 996b29d

Please sign in to comment.