ashbhandare 85434273ff Fix CUDA Reduction kernel for ArgMax/ArgMix for when reduction dim=1 (#6490) ... * Fix for when reduction dim=1 * Disable test for AMD GPUs * Specify Async		2021-02-02 09:50:16 -08:00
..
github	Fix CUDA Reduction kernel for ArgMax/ArgMix for when reduction dim=1 (#6490)	2021-02-02 09:50:16 -08:00
__init__.py	Add validation of op registrations (#5817)	2020-11-17 10:44:09 -08:00
amd_hipify.py	Enable dense sequence optimized version of Pytorch exported BERT-L on AMD GPU (#6504)	2021-01-29 13:12:34 -08:00
build.py	Introduce --enable_training_ops build flag (#6523)	2021-02-01 21:54:16 -08:00
clean_docker_image_cache.py	Fix clean_docker_image_cache.py detection of image pushes. (#6151)	2020-12-16 17:25:22 -08:00
coverage.py	Add support for running Android emulator from build.py on Windows. (#6317)	2021-01-13 19:21:49 -08:00
exclude_unused_ops_and_types.py	Add ability to track per operator types in reduced build config. (#6428)	2021-01-29 07:59:51 +10:00
gen_def.py	Reduce IOS shared library size by symbol file. (#5171)	2020-09-14 23:59:41 -07:00
get_docker_image.py	Update get_docker_image.py to enable use without image cache container registry. (#6177)	2020-12-18 19:01:02 -08:00
logger.py	Cache build docker images in container registry. (#5811)	2020-11-17 17:02:24 -08:00
op_registration_utils.py	Add ability to track per operator types in reduced build config. (#6428)	2021-01-29 07:59:51 +10:00
op_registration_validator.py	Cache build docker images in container registry. (#5811)	2020-11-17 17:02:24 -08:00