| .. |
|
codegen
|
Clean up profiling mode and profiling executor strategy (#73875)
|
2022-03-29 18:38:51 +00:00 |
|
data
|
|
|
|
distributed
|
Disable RPC profiling for kineto profilers
|
2022-04-26 23:35:30 +00:00 |
|
generated
|
|
|
|
test_module
|
|
|
|
__init__.py
|
|
|
|
autocast_test_lists.py
|
Add Autocast support for Einsum (#71916)
|
2022-03-08 22:04:30 +00:00 |
|
check_kernel_launches.py
|
|
|
|
common_cuda.py
|
More distutils.version.LooseVersion changes (#69947)
|
2021-12-15 08:07:36 -08:00 |
|
common_device_type.py
|
Various OpInfo architecture improvements
|
2022-04-18 21:55:32 +00:00 |
|
common_distributed.py
|
[Distributed tests] Add skip for odd world_size condition
|
2022-04-24 22:04:30 +00:00 |
|
common_dtype.py
|
|
|
|
common_fsdp.py
|
[FSDP] Add ignored_modules ctor arg
|
2022-04-12 19:46:00 +00:00 |
|
common_fx2trt.py
|
[fx2trt] support ops for hf_T5 (#62)
|
2022-04-27 00:38:11 +00:00 |
|
common_jit.py
|
Extends OpInfo architecture with reference inputs, adds them for elementwise binary operators
|
2022-03-21 03:24:16 +00:00 |
|
common_methods_invocations.py
|
remove unneeded overload for nansum
|
2022-04-27 16:02:54 +00:00 |
|
common_modules.py
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +00:00 |
|
common_nn.py
|
Implement Tanh Gelu Approximation (#61439)
|
2022-02-14 03:40:32 +00:00 |
|
common_quantization.py
|
add channels last (2d) support for mkldnn_convolution (#55584)
|
2022-04-20 22:34:44 +00:00 |
|
common_quantized.py
|
Add onednn quant backend (#74137)
|
2022-03-15 01:28:21 +00:00 |
|
common_subclass.py
|
Support for tensor subclasses as parameters
|
2022-04-27 19:28:55 +00:00 |
|
common_utils.py
|
[primTorch] Prototype tracer and elementwise unary reference opinfo class
|
2022-04-27 14:40:21 +00:00 |
|
composite_compliance.py
|
[composite compliance] add test for fwd AD
|
2022-04-25 15:15:48 +00:00 |
|
dist_utils.py
|
|
|
|
hypothesis_utils.py
|
|
|
|
jit_metaprogramming_utils.py
|
[JIT] Reuse traced fn for jit opinfos
|
2022-04-22 20:14:29 +00:00 |
|
jit_utils.py
|
Clean up profiling mode and profiling executor strategy (#73875)
|
2022-03-29 18:38:51 +00:00 |
|
logging_tensor.py
|
Support for tensor subclasses as parameters
|
2022-04-27 19:28:55 +00:00 |
|
opinfo_helper.py
|
[opinfo] use dtypes instead of dtypesIfCPU (#68732)
|
2021-11-27 16:07:51 -08:00 |
|
quantization_torch_package_models.py
|
|
|
bc34cf5fe4