Newsha Ardalani 0fb58d76a1 Support ArgMin in c2_pt_converter ...

Summary:
+ Add ArgMin support to Caffe2 to PyTorch converter
+ Using hypothesis to parameterize different conditions for test

Test Plan: buck test //caffe2/torch/fb/model_transform/c2_convert:c2_pt_converter_test

Reviewed By: houseroad

Differential Revision: D25016203

fbshipit-source-id: 94489fcf1ed3183ec96f9796a5b4fb348fbde5bc

2020-12-05 16:35:34 -08:00

..

__init__.py

remediation of S205607

2020-07-17 17:19:47 -07:00

algebra.py

Support ArgMin in c2_pt_converter

2020-12-05 16:35:34 -08:00

arg_scope.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

array_helpers.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

control_ops.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

conv.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

db_input.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

dropout.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

elementwise_linear.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

fc.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

nonlinearity.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

normalization.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

pooling.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

quantization.py

Fused8BitRowwiseQuantizedToFloat operator support (#48407)

2020-11-30 17:11:39 -08:00

tools.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00

train.py

Remove __future__ imports for legacy Python2 supports (#45033)

2020-09-23 17:57:02 -07:00