strisunshinewentingwang/pytorch

Pull Requests Commits

DNNL: enable dilation conv (#40220)

XiaobingSuper committed 4 years ago

dbcc5b75

Port addmm, addbmm, addr to ATen (CUDA) (#38421)

Baranowski committed 4 years ago

43331609

DNNL: enable max_pool3d and avg_pool3d (#35664)

XiaobingSuper committed 4 years ago

c8738957

DNNL: enable batchnorm3d (#35663)

XiaobingSuper committed 4 years ago

8df35fd7

DNNL: enable conv3d (#35662)

XiaobingSuper committed 4 years ago

6ba807cb

Utilise the vector version for sinh and cosh (UnaryOpsKernel) (#36396)

kshitij12345 committed 4 years ago

03af4dcb

jit: Conv3d + BatchNorm3d fusion (#40082)

XiaobingSuper committed 4 years ago

87c5f02f

Add prefix of remote events for RPC profiling (#40066)

rohan-varma committed 4 years ago

14f7e95c

Relax cudnn conditions for channels-last convolutions (#38904)

xwang233 committed 4 years ago

17d3f74e

Add addmv for complex dtypes (#40238)

anjali411 committed 4 years ago

c72ab194

Reenable memory format test for some unary functions (#39102)

xuhdev committed 4 years ago

3894de56

[quant][refactor] Tests for torch.jit.quantized (#40330)

jerryzh168 committed 4 years ago

9f9e7c1d

[ONNX] Support large attribute and subgraph for large model (#38793)

BowenBao committed 4 years ago

eaa91071

reference Swish implementation (#40150)

Hector Yuen committed 4 years ago

49887d1f

ONNX: fix bug in export of cumsum operator (#40044)

yaeldMS committed 4 years ago

3fa0b1e3

[quant][bug] Histogram observer bug fix with min == max (#40310)

supriyar committed 4 years ago

c04d39aa

ONNX: fix bug in export of ops involving torch.bool type (#40006)

yaeldMS committed 4 years ago

766889b6

Update TensorPipe submodule (#40374)

lw committed 4 years ago

0e146d2d

Meta tensors, but without code deduplication (#38490)

ezyang committed 4 years ago

e4766fb4

ROCm: Use correct device type when exporting tensors to DLPack (#40124)

t-vi committed 4 years ago

52f3a096

Rename dont_resize_outputs() to resize_outputs(false) (TensorIterator… (#40148)

guol-fnst committed 4 years ago

db5b2739

[ROCm] Enable BFloat16 for pow, exp, erf ops on ROCm (#40236)

rohithkrn committed 4 years ago

396087bf

Fixed buffer update in BatchNorm when track_running_stats is set to False (#38084)

frgfm committed 4 years ago

881c1adf

Append forward slashes to PIP_UPLOAD_FOLDER (#40352)

peterjc123 committed 4 years ago

eb92ed62

Pin the version of scipy for Windows test jobs (#40369)

peterjc123 committed 4 years ago

37c88a47

graph mode: add hardswish inplace handling (#40284)

vkuzo committed 4 years ago

ab8a99bd

quantized elu: graph mode handling (#40111)

vkuzo committed 4 years ago

c6dbfcaf

quantized elu: eager mode QAT handling (#40104)

vkuzo committed 4 years ago

cd0afe2b

quantized elu: eager mode static handling (#40103)

vkuzo committed 4 years ago

03ed802a

quantized elu: require observation (#40100)

vkuzo committed 4 years ago

13d54c64

Newer Older