pytorch/test/cpp
Mikhail Zolotukhin 75ce040620 [TensorExpr] Allow for 'keepdim' argument in aten::mean in NNC's external call. (#68756)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68756

That fixes some warnings in our tests.

Test Plan: Imported from OSS

Reviewed By: navahgar

Differential Revision: D32600952

Pulled By: ZolotukhinM

fbshipit-source-id: 548eaf3659e20795cce44d8f57e77f4a47d44d98
2021-11-30 00:06:34 -08:00
..
api Add set_to_none option to C++ API (#68801) 2021-11-29 08:42:39 -08:00
c10d Enable desync root cause analysis for NCCL (#68310) 2021-11-17 20:29:03 -08:00
common
dist_autograd
jit Populate operator_input_sizes_ (#68542) 2021-11-23 12:18:06 -08:00
lazy Add lazy::Shape::numel() (#68314) 2021-11-29 08:38:09 -08:00
lite_interpreter_runtime Back out "Revert D30710710: [Pytorch Edge] Support profiling kineto events from external source" (#66421) 2021-10-12 10:55:29 -07:00
monitor torch.monitor - Initial C++ Stats (#68074) 2021-11-18 21:46:23 -08:00
rpc Remove ProcessGroup from TensorPipeAgent initialization (#68128) 2021-11-11 12:28:55 -08:00
tensorexpr [TensorExpr] Allow for 'keepdim' argument in aten::mean in NNC's external call. (#68756) 2021-11-30 00:06:34 -08:00
__init__.py