OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
84427d26db
pytorch/caffe2/python/operator_test
History
Xiaomeng Yang 87cac4c2f1 Update Im2Col related to make preparation for group conv in NHWC order. (#10439) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/10439

Update Im2Col related to make preparation for group conv in NHWC order.

Reviewed By: houseroad

Differential Revision: D9285344

fbshipit-source-id: 1377b0243acb880d2ad9cf73084529a787dcb97d
2018-08-15 17:10:24 -07:00
..
activation_ops_test.py Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
adadelta_test.py Add Adadelta optimizer to caffe2 (#9088) 2018-07-24 20:09:21 -07:00
adagrad_test_helper.py Update from facebook 1ee4edd286a3 (#8040) 2018-06-01 17:41:09 -04:00
adagrad_test.py Update from facebook 1ee4edd286a3 (#8040) 2018-06-01 17:41:09 -04:00
adam_test.py Update from facebook (#7696) 2018-05-19 23:10:48 -07:00
affine_channel_op_test.py Add implementation for batch_moments_op (#9510) 2018-07-18 11:59:54 -07:00
apmeter_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
arg_ops_test.py [caffe2] Update math functions with params on host. (#6602) 2018-04-14 21:41:41 -07:00
assert_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
atomic_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
basic_rnn_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_box_cox_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_bucketize_op_test.py Add BatchBucketizeOp in caffe2 (#9385) 2018-07-13 20:39:30 -07:00
batch_moments_op_test.py Add implementation for batch_moments_op (#9510) 2018-07-18 11:59:54 -07:00
batch_sparse_to_dense_op_test.py Add batch-based/row-based sparse from/to dense operator 2017-11-16 00:59:21 -08:00
bbox_transform_test.py Clip horizontal bounding boxes during rotated detection for backward compatibility (#9403) 2018-07-16 20:24:49 -07:00
blobs_queue_db_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
boolean_mask_test.py BBoxTransform op: Add support for rotated boxes (#8952) 2018-07-11 10:25:34 -07:00
boolean_unmask_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
box_with_nms_limit_op_test.py Fixed a bug in box_with_nms_limit where it may produce more bounding boxes than specified. (#10390) 2018-08-14 23:54:23 -07:00
cast_op_test.py cast op for empty batch 2017-11-16 12:20:20 -08:00
ceil_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
channel_backprop_stats_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
channel_shuffle_test.py Update channel_shuffle_op and transpose 2d to speed up ShuffleNet (#9525) 2018-07-21 12:54:33 -07:00
channel_stats_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
checkpoint_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
clip_op_test.py Add Hardtanh Export (#8804) 2018-07-11 18:09:51 -07:00
clip_tensor_op_test.py Add gradient clipping (#2452) 2018-03-27 15:10:15 -07:00
collect_and_distribute_fpn_rpn_proposals_op_test.py [Caffe2] Fix the wrong argument name in collect_and_distribute_op (#7091) 2018-04-30 15:01:11 -07:00
concat_split_op_test.py [Caffe2] Fix gradient_check on in-place ops (#8828) 2018-06-25 15:25:56 -07:00
conditional_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_test.py Update Im2Col related to make preparation for group conv in NHWC order. (#10439) 2018-08-15 17:10:24 -07:00
conv_transpose_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
copy_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
cosine_embedding_criterion_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
counter_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
crf_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
cross_entropy_ops_test.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
ctc_beam_search_decoder_op_test.py Add ctc_beam_search_decoder op for caffe2 (#9622) 2018-07-23 13:40:24 -07:00
ctc_greedy_decoder_op_test.py Fix default argument value for CTCGreedyDecoder op (#9747) 2018-07-27 16:33:07 -07:00
cudnn_recurrent_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
dataset_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
deform_conv_test.py Deformable convolution operator in Caffe2 2017-09-26 16:20:31 -07:00
depthwise_3x3_conv_test.py [caffe2] Open-source DEPTHWISE_3x3 engine (#6601) 2018-04-26 02:30:51 -04:00
distance_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
dropout_op_test.py Eanble python tests on ROCM (#9616) 2018-07-24 11:37:58 -07:00
duplicate_operands_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
elementwise_linear_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
elementwise_logical_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
elementwise_op_broadcast_test.py Update from facebook (#8384) 2018-06-13 13:10:45 -07:00
elementwise_ops_test.py Adding reciprocal operator and a test 2018-07-27 18:24:43 -07:00
emptysample_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
enforce_finite_op_test.py EnforceFinite test (#10143) 2018-08-03 10:31:29 -07:00
ensure_clipped_test.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
ensure_cpu_output_op_test.py Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
expand_op_test.py In Expand support using -1 to indicate preserving original size (#10174) 2018-08-02 22:09:47 -07:00
extend_tensor_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
fc_operator_test.py Disable "filter too much" health check for fc operator tests (#9865) 2018-07-25 21:41:14 -07:00
feature_maps_ops_test.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
filler_ops_test.py Update from facebook (#8384) 2018-06-13 13:10:45 -07:00
find_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
flatten_op_test.py Enable Flatten operator to take an arbitrary axis arguemnt 2017-10-13 12:28:22 -07:00
flexible_top_k_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
floor_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
gather_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
gather_ranges_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
given_tensor_byte_string_to_uint8_fill_op_test.py Export uint8 tensors as byte string in mobile_exporter and add GivenTensorByteStringToUInt8FillOp (#10385) 2018-08-15 14:26:50 -07:00
given_tensor_fill_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
glu_op_test.py Supress hypothesis health check in glu_op_test.py 2017-12-03 22:51:46 -08:00
group_conv_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
group_norm_op_test.py Update channel_shuffle_op and transpose 2d to speed up ShuffleNet (#9525) 2018-07-21 12:54:33 -07:00
gru_test.py [Caffe2] Fix gradient_check on in-place ops (#8828) 2018-06-25 15:25:56 -07:00
hsm_test.py [Caffe2] Fix gradient_check on in-place ops (#8828) 2018-06-25 15:25:56 -07:00
hyperbolic_ops_test.py Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
im2col_col2im_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
image_input_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
index_hash_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
index_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
instance_norm_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
integral_image_ops_test.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
jsd_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
key_split_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
lars_test.py Improve lars operator by applying clipping (#9905) 2018-08-02 11:54:28 -07:00
layer_norm_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
leaky_relu_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
learning_rate_adaption_op_test.py Update from facebook (#7696) 2018-05-19 23:10:48 -07:00
learning_rate_op_test.py Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
lengths_pad_op_test.py Update from facebook (#6871) 2018-04-23 15:01:56 -07:00
lengths_tile_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
lengths_top_k_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
listwise_l2r_operator_test.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
load_save_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
locally_connected_op_test.py Update Im2ColNd functions (#7505) 2018-05-12 15:59:50 -07:00
loss_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
lpnorm_op_test.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
map_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
margin_ranking_criterion_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
math_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
matmul_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mean_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
merge_id_lists_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mkl_conv_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mkl_packed_fc_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mkl_speed_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mod_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
moments_op_test.py Fix lint errors (#7247) 2018-05-03 12:17:23 -07:00
momentum_sgd_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mpi_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
negate_gradient_op_test.py NegateGradientOp and test 2017-11-13 18:05:14 -08:00
ngram_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
normalize_op_test.py Fix l2 normalization when handling zero vector (#9594) 2018-07-19 14:10:03 -07:00
numpy_tile_op_test.py [caffe2][ONNX] Implement CPU NumpyTileOp and corresponding ONNX backend (#7053) 2018-04-27 19:58:15 -07:00
one_hot_ops_test.py Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
onnx_while_test.py [caffe2] Add flag to ONNXWhile to skip scoping (#6910) 2018-04-24 16:53:22 -07:00
order_switch_test.py generalize order switch ops for 1-3d (#10395) 2018-08-15 10:09:31 -07:00
pack_ops_test.py Fix the error in UnpackSegmentsOp when calculating the gradient with "max_length" argument (#9598) 2018-07-20 11:09:34 -07:00
pack_rnn_sequence_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
pad_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
partition_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
percentile_op_test.py Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
piecewise_linear_transform_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
pooling_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
prepend_dim_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
python_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rand_quantization_op_test.py adding stochastic quantization caffe2 operators (encoder and decoder in CPU are implemented. GPU mode is pending) 2018-08-13 16:39:23 -07:00
rank_loss_operator_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rebatching_queue_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
record_queue_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
recurrent_net_executor_test.py Potential fix for RNN test on MKL (#7862) 2018-05-25 16:16:46 -07:00
recurrent_network_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
reduce_ops_test.py Add support for .norm() pytorch onnx export and ReduceL1/ReduceL2 caffe2 operators (#9299) 2018-07-14 10:54:13 -07:00
reduction_ops_test.py Add SumElementsInt operator 2018-03-27 18:10:39 -07:00
reshape_ops_test.py Fix dividing by zero segfault in Reshape (#8302) 2018-06-09 09:48:22 -07:00
resize_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rmac_regions_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rnn_cell_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
roi_align_rotated_op_test.py Make roi_align_rotated_op_test not rely on 1.12.0 numpy.rot90 (#9267) 2018-07-09 11:55:39 -07:00
segment_ops_test.py Update from facebook (#7696) 2018-05-19 23:10:48 -07:00
selu_op_test.py selu op 2017-12-15 15:38:44 -08:00
sequence_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
shape_inference_test.py Add interface to provide blob types to shape&type inference (#9643) 2018-07-24 11:58:05 -07:00
sinusoid_position_encoding_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
softmax_ops_test.py fix expty batch for softmax (#9075) 2018-07-01 16:40:14 -07:00
softplus_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
sparse_gradient_checker_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
sparse_lengths_sum_benchmark.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
sparse_normalize_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
sparse_ops_test.py Eanble python tests on ROCM (#9616) 2018-07-24 11:37:58 -07:00
sparse_to_dense_mask_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
spatial_bn_op_test.py Update hip files (#9826) 2018-07-27 16:54:39 -07:00
specialized_segment_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
square_root_divide_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
stats_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
string_ops_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
text_file_reader_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
thresholded_relu_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
tile_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
top_k_test.py Update from facebook (#6871) 2018-04-23 15:01:56 -07:00
transpose_op_test.py [caffe2] Update math functions with params on host. (#6602) 2018-04-14 21:41:41 -07:00
trigonometric_op_test.py Fix trigonometric_op_test failures when running in python3.6 (#7831) 2018-05-24 19:09:35 -07:00
unique_ops_test.py [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -07:00
unique_uniform_fill_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
upsample_op_test.py bilinear upsample operator on CPU 2018-06-27 10:12:06 -07:00
utility_ops_test.py Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
video_input_op_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
weighted_multi_sample_test.py Only run WeightedMultiSample test on CPU and not GPU. 2018-03-20 13:34:22 -07:00
weighted_sample_test.py support non-normalized weights 2017-10-27 23:18:25 -07:00
weighted_sum_test.py Don't run CUDA tests for ops without CUDA implementation 2017-11-08 10:28:02 -08:00
wngrad_test.py Implementation and operator test for Wngrad optimizer (#8999) 2018-07-13 18:11:41 -07:00