pytorch/caffe2/python/helpers
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
algebra.py Support ArgMin in c2_pt_converter 2020-12-05 16:35:34 -08:00
arg_scope.py
array_helpers.py
control_ops.py
conv.py
db_input.py
dropout.py
elementwise_linear.py
fc.py
nonlinearity.py
normalization.py
pooling.py
quantization.py Fused8BitRowwiseQuantizedToFloat operator support (#48407) 2020-11-30 17:11:39 -08:00
tools.py
train.py