File tree Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -21,8 +21,10 @@ def bench():
21
21
reduced = np .max (reduced , - 1 )
22
22
else :
23
23
reduced = np .min (reduced , - 1 )
24
+ print (reduced )
24
25
return reduced
25
- red = tacoBench (bench )
26
+ tacoBench (bench )
27
+ red = bench ()
26
28
print ("reduced value for" , dims , "is" , red )
27
29
28
30
@pytest .mark .parametrize ("dims" , [1 , 3 , 5 ])
Original file line number Diff line number Diff line change @@ -453,10 +453,9 @@ def construct_minmax_tensor_key(dims, variant=None):
453
453
return os .path .join (path , name , key )
454
454
455
455
def generate_crds_helper (shape , level , crds ):
456
- sampling = 0.1
457
- num = 3
458
- std = 2
459
- last_layer_sampling = 0.4
456
+ sampling = 0.4
457
+ num = 4
458
+ std = 2
460
459
461
460
if level == len (shape ) - 1 :
462
461
return crds
You can’t perform that action at this time.
0 commit comments