Skip to content

Commit

Permalink
add validation flag to layer norm test
Browse files Browse the repository at this point in the history
  • Loading branch information
yardeny-sony committed Sep 25, 2024
1 parent f8edb99 commit 9116ae7
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,7 @@ def __init__(self, unit_test, has_weight=None, has_bias=None):
super().__init__(unit_test)
self.has_weight = has_weight
self.has_bias = has_bias
self.use_is_close_validation = True

def create_inputs_shape(self):
return [[self.val_batch_size, 3, 32, 32]]
Expand Down

0 comments on commit 9116ae7

Please sign in to comment.