mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
c755616e00
Summary: 1. Support ops needed for inference of Faster-RCNN/Mask-RCNN needed in Detectron, mostly direct fallbacks. 2. Use CPU device to hold 0-dim tensors and integer tensors in both fallback op and blob feeder, needed by Detectron models. 3. Ignore 0-dim tensor in MKL-DNN concat operator. 4. Generate dynamic library of Detectron module for CPU device. This PR obsoletes #9164. Pull Request resolved: https://github.com/pytorch/pytorch/pull/10157 Differential Revision: D9276837 Pulled By: yinghai fbshipit-source-id: dc364932ae4a2e7fcefdee70b5fce3c0cee91b6f |
||
|---|---|---|
| .. | ||
| concat_split_op_test.py | ||
| conv_op_test.py | ||
| convfusion_op_test.py | ||
| copy_op_test.py | ||
| dropout_op_test.py | ||
| elementwise_sum_op_test.py | ||
| fc_op_test.py | ||
| LRN_op_test.py | ||
| operator_fallback_op_test.py | ||
| pool_op_test.py | ||
| relu_op_test.py | ||
| softmax_op_test.py | ||
| spatial_bn_op_test.py | ||
| squeeze_op_test.py | ||
| test_ideep_net.py | ||
| transform_ideep_net.py | ||