Skip to content

Commit

Permalink
#0: testing
Browse files Browse the repository at this point in the history
  • Loading branch information
shwetankTT committed Dec 6, 2024
1 parent 009aecf commit 9b2af82
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ def test_prepare_conv_weights(
tt_weight_tensor_formatted = ttnn.to_device(tt_weight_tensor_formatted, device)
tt_bias_tensor_formatted = ttnn.to_device(tt_bias_tensor_formatted, device) if has_bias else None
(k := next(iter(conv_kwargs)), conv_kwargs.pop(k)) ##removing 1st element from dict
tt_output_tensor_on_device, _, _, _, _ = ttnn.conv2d(
tt_output_tensor_on_device = ttnn.conv2d(
input_tensor=tt_input_tensor,
weight_tensor=tt_weight_tensor_formatted,
bias_tensor=tt_bias_tensor_formatted,
Expand Down

0 comments on commit 9b2af82

Please sign in to comment.