chenyu 820f2e054e fix PADTO optimization (#2935) ... the correct condition is that PADTO cannot be applied to reduce axis, not Reduce.MAX in ops. even for Reduce.SUM it's possible that the reduce axis had a div before, and the padded 0 became inf then sum over it is incorrect.		2023-12-25 22:52:49 -05:00
..
extract_policynet.py	merge kernel and optimizer (#2200)	2023-11-01 15:20:01 -07:00
extract_sa_pairs.py	merge kernel and optimizer (#2200)	2023-11-01 15:20:01 -07:00
generate_dataset.sh	op logger + replay (#2021)	2023-10-08 15:10:18 -07:00
get_action_space.py	merge kernel and optimizer (#2200)	2023-11-01 15:20:01 -07:00
helpers.py	green linearizer asserts for ops (#2800)	2023-12-25 10:41:54 -05:00
pretrain_valuenet.py	merge kernel and optimizer (#2200)	2023-11-01 15:20:01 -07:00
rl.py	fixed imports (#2185)	2023-10-30 22:07:17 -07:00
run_qnet.py	Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)	2023-10-27 10:53:06 -10:00
test_beam_search.py	fix PADTO optimization (#2935)	2023-12-25 22:52:49 -05:00
test_net.py	fixed imports (#2185)	2023-10-30 22:07:17 -07:00
test_time_linearizer.py	KOPT is over, BEAM is upstream (#2071)	2023-10-16 09:46:03 -07:00