pytorch/test/cpp
Zino Benaissa 40c77f926c Add prim::TypeCheck operation (#43026)
Summary:
TypeCheck is a new operation to check the shape of tensors against
 expectd shapes. TypeCheck is a variadic operation. An example,

 %t0 : Tensor = ...
 %t1 : Tensor = ...
 %2 : FLOAT(20, 20), %3 : FLOAT(30, 30), %1 : bool =
 prim::TypeCheck(%t1, %t2)
 prim::If(%1)

Fixes #{issue number}

Pull Request resolved: https://github.com/pytorch/pytorch/pull/43026

Reviewed By: ZolotukhinM

Differential Revision: D23115830

Pulled By: bzinodev

fbshipit-source-id: fbf142126002173d2d865cf4b932dea3864466b4
2020-08-21 20:03:24 -07:00
..
api remove empty override pretty_print (#43341) 2020-08-20 18:48:29 -07:00
common
dist_autograd Release DistAutogradContainer context for each dist_autograd test case (#38711) 2020-06-18 20:58:55 -07:00
jit Add prim::TypeCheck operation (#43026) 2020-08-21 20:03:24 -07:00
rpc Add set_device_map to TensorPipeOptions to support GPU args (#42637) 2020-08-14 18:46:55 -07:00
tensorexpr Normalize loops with non-zero start. (#43179) 2020-08-21 12:37:27 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00