| .. |
|
__init__.py
|
remediation of S205607
|
2020-07-17 17:19:47 -07:00 |
|
activation_ops_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
adadelta_test.py
|
eliminate FE_INVALID in optimizer related operators and tests (#20501)
|
2019-05-16 08:23:46 -07:00 |
|
adagrad_test.py
|
implement L2 regularization for Adagrad in caffe2 and dper (#37705)
|
2020-05-03 10:42:49 -07:00 |
|
adagrad_test_helper.py
|
[caffe2] weight_decay in reduced precision adagrad
|
2020-05-12 20:33:40 -07:00 |
|
adam_test.py
|
Support RowWiseSparseAdam on GPU (#35404)
|
2020-07-31 10:47:29 -07:00 |
|
affine_channel_op_test.py
|
|
|
|
alias_with_name_test.py
|
move AliasWithNameOp to caffe2/operators
|
2019-12-17 02:39:40 -08:00 |
|
apmeter_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
arg_ops_test.py
|
|
|
|
assert_test.py
|
|
|
|
atomic_ops_test.py
|
Add 64bit atomic fetch add (#32354)
|
2020-01-17 11:43:43 -08:00 |
|
basic_rnn_test.py
|
|
|
|
batch_box_cox_test.py
|
|
|
|
batch_bucketize_op_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
batch_moments_op_test.py
|
|
|
|
batch_sparse_to_dense_op_test.py
|
Add cuda version for operators BatchSparseToDense and BatchDenseToSparse (#29166)
|
2019-11-05 13:06:23 -08:00 |
|
bbox_transform_test.py
|
|
|
|
bisect_percentile_op_test.py
|
|
|
|
blobs_queue_db_test.py
|
caffe2 - set up correct inheritance structure for remaining operator test classes (#18622)
|
2019-04-01 15:53:22 -07:00 |
|
boolean_mask_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
boolean_unmask_test.py
|
|
|
|
box_with_nms_limit_op_test.py
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
bucketize_op_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
cast_op_test.py
|
|
|
|
ceil_op_test.py
|
|
|
|
channel_backprop_stats_op_test.py
|
|
|
|
channel_shuffle_test.py
|
batch size 0 support in ChannelShuffle DNNLOWP op (#26858)
|
2019-09-26 00:40:07 -07:00 |
|
channel_stats_op_test.py
|
Optimize channel_stats_op (#16243)
|
2019-03-12 12:08:00 -07:00 |
|
checkpoint_test.py
|
Skip DataIO tests relying on LevelDB if compiled without it (#42169)
|
2020-07-28 10:18:26 -07:00 |
|
clip_op_test.py
|
|
|
|
clip_tensor_op_test.py
|
|
|
|
collect_and_distribute_fpn_rpn_proposals_op_test.py
|
split and register CollectAndDistributeFpnRpnProposals with C10
|
2019-05-16 13:40:46 -07:00 |
|
concat_split_op_test.py
|
[caffe2] fix type and shape inference for common gradient ops (#35857)
|
2020-04-02 11:17:04 -07:00 |
|
conditional_test.py
|
|
|
|
conftest.py
|
|
|
|
conv_test.py
|
[ROCm] Enable 3D convolutions through ROCm (#33067)
|
2020-02-14 13:19:10 -08:00 |
|
conv_transpose_test.py
|
fix zero-batch handling in convtranspose (#24341)
|
2019-12-18 15:06:36 -08:00 |
|
copy_ops_test.py
|
caffe2 - set up correct inheritance structure for remaining operator test classes (#18622)
|
2019-04-01 15:53:22 -07:00 |
|
copy_rows_to_tensor_op_test.py
|
Perform weight re-init for embedding table in sparse_lookup.py (#22348)
|
2019-07-03 10:33:40 -07:00 |
|
cosine_embedding_criterion_op_test.py
|
|
|
|
counter_ops_test.py
|
|
|
|
crf_test.py
|
[Python] Deprecate use of scipy.misc.logsumexp and scipy.misc.comb (#32209)
|
2020-01-15 10:40:47 -08:00 |
|
cross_entropy_ops_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
ctc_beam_search_decoder_op_test.py
|
Output sequence probability with CTC beam search, optional multiple output sequences (#21927)
|
2019-07-02 17:29:13 -07:00 |
|
ctc_greedy_decoder_op_test.py
|
|
|
|
cudnn_recurrent_test.py
|
Unify gpu_support variable in python tests (#16748)
|
2019-02-07 00:29:51 -08:00 |
|
data_couple_op_test.py
|
|
|
|
dataset_ops_test.py
|
[caffe2] Add support for std::shared_ptr<std::vector<TensorList>> in PackRecordsOp and UnPackRecordsOp (#36550)
|
2020-04-14 03:43:21 -07:00 |
|
deform_conv_test.py
|
no EIGEN engine for DeformConv (#16785)
|
2019-02-06 11:59:31 -08:00 |
|
dense_vector_to_id_list_op_test.py
|
|
|
|
depthwise_3x3_conv_test.py
|
|
|
|
detectron_keypoints.py
|
|
|
|
distance_op_test.py
|
|
|
|
dropout_op_test.py
|
|
|
|
duplicate_operands_test.py
|
|
|
|
elementwise_linear_op_test.py
|
|
|
|
elementwise_logical_ops_test.py
|
|
|
|
elementwise_op_broadcast_test.py
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
elementwise_ops_test.py
|
Mish Activation Function (#40856)
|
2020-07-06 15:51:23 -07:00 |
|
emptysample_ops_test.py
|
|
|
|
enforce_finite_op_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
ensure_clipped_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
ensure_cpu_output_op_test.py
|
|
|
|
erf_op_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
expand_op_test.py
|
|
|
|
fc_operator_test.py
|
Fix spelling errors (#21665)
|
2019-06-13 15:21:55 -07:00 |
|
feature_maps_ops_test.py
|
Add a new op for converting the dense feature to sparse representation
|
2020-07-27 12:45:37 -07:00 |
|
filler_ops_test.py
|
|
|
|
find_op_test.py
|
|
|
|
flatten_op_test.py
|
|
|
|
flexible_top_k_test.py
|
|
|
|
floor_op_test.py
|
|
|
|
fused_nbit_rowwise_conversion_ops_test.py
|
[caffe2] optimize 2/4-bit row-wise quantization (#387)
|
2020-06-19 21:28:31 -07:00 |
|
fused_nbit_rowwise_test.cc
|
[caffe2] open source 2/4-bit SLS operators (#34903)
|
2020-03-17 22:55:10 -07:00 |
|
fused_nbit_rowwise_test_helper.py
|
[caffe2] open source 2/4-bit SLS operators (#34903)
|
2020-03-17 22:55:10 -07:00 |
|
gather_ops_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
gather_ranges_op_test.py
|
[fix][tiny][caffe2] Avoid triggering errors when allow ratio is 100% (#34757)
|
2020-03-16 11:39:32 -07:00 |
|
given_tensor_byte_string_to_uint8_fill_op_test.py
|
|
|
|
given_tensor_fill_op_test.py
|
Add GivenTensorInt16Fill (#20515)
|
2019-05-15 19:45:15 -07:00 |
|
glu_op_test.py
|
|
|
|
group_conv_test.py
|
omit group conv NHWC test for GPU (#17715)
|
2019-03-06 11:32:35 -08:00 |
|
group_norm_op_test.py
|
Optimize channel_stats_op (#16243)
|
2019-03-12 12:08:00 -07:00 |
|
gru_test.py
|
|
|
|
heatmap_max_keypoint_op_test.py
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
histogram_test.py
|
[AMD] Fix broken test (#39297)
|
2020-05-30 13:12:24 -07:00 |
|
hsm_test.py
|
|
|
|
hyperbolic_ops_test.py
|
[caffe2] fix type and shape inference for common gradient ops (#35857)
|
2020-04-02 11:17:04 -07:00 |
|
im2col_col2im_test.py
|
|
|
|
image_input_op_test.py
|
|
|
|
index_hash_ops_test.py
|
Support in-place update in IndexHashOp (#30275)
|
2019-11-22 14:49:28 -08:00 |
|
index_ops_test.py
|
|
|
|
instance_norm_test.py
|
Optimize InstanceNormGradientOp (#22288)
|
2019-07-01 15:10:17 -07:00 |
|
integral_image_ops_test.py
|
|
|
|
jsd_ops_test.py
|
|
|
|
key_split_ops_test.py
|
|
|
|
lars_test.py
|
Add missing shebangs to Python files with executable permissions.
|
2019-06-06 10:53:40 -07:00 |
|
layer_norm_op_test.py
|
Add elementwise_affine for LayerNormGradientOp (#19982)
|
2019-05-03 15:33:46 -07:00 |
|
leaky_relu_test.py
|
|
|
|
learning_rate_adaption_op_test.py
|
|
|
|
learning_rate_op_test.py
|
[GradualGating] support better k value change (#41557)
|
2020-07-17 20:44:28 -07:00 |
|
length_split_op_test.py
|
|
|
|
lengths_pad_op_test.py
|
|
|
|
lengths_reducer_fused_nbit_rowwise_ops_test.py
|
Split SparseLengthsSumSparse into SparseLengthsSumSparseLookup + SparseLengthsSum (#35507)
|
2020-03-30 13:33:29 -07:00 |
|
lengths_tile_op_test.py
|
|
|
|
lengths_top_k_ops_test.py
|
|
|
|
listwise_l2r_operator_test.py
|
add LambdaRank DCG Loss Option (#23679)
|
2019-08-02 11:47:46 -07:00 |
|
load_save_test.py
|
Avoid Output Uninitialized Blobs in Load with load_all=1 (#19133)
|
2019-04-27 10:45:44 -07:00 |
|
locally_connected_op_test.py
|
|
|
|
loss_ops_test.py
|
|
|
|
lpnorm_op_test.py
|
Back out "Back out "[c2] register cuda op for LpNorm (fallback)"" (#38566)
|
2020-05-19 10:37:25 -07:00 |
|
map_ops_test.py
|
|
|
|
margin_ranking_criterion_op_test.py
|
|
|
|
math_ops_test.py
|
[Caffe2] Fix shape inference for element-wise operators (#33431)
|
2020-02-25 09:03:06 -08:00 |
|
matmul_op_test.py
|
[caffe2] remove unnecessary np.set_printoptions and fix test errors (#32475)
|
2020-01-22 14:49:47 -08:00 |
|
mean_op_test.py
|
|
|
|
merge_id_lists_op_test.py
|
|
|
|
mkl_conv_op_test.py
|
|
|
|
mkl_packed_fc_op_test.py
|
|
|
|
mod_op_test.py
|
|
|
|
moments_op_test.py
|
|
|
|
momentum_sgd_test.py
|
Disable flaky TestMomentumSGD.test_fp16momentum_sgd (#31369)
|
2019-12-17 19:16:54 -08:00 |
|
mpi_test.py
|
|
|
|
mul_gradient_benchmark.py
|
optimize MulGradient for common shapes (#19705)
|
2019-12-11 11:39:52 -08:00 |
|
negate_gradient_op_test.py
|
|
|
|
ngram_ops_test.py
|
|
|
|
normalize_op_test.py
|
Revert D22330340: [C2] Fixed a bug in normalization operator
|
2020-07-02 16:05:23 -07:00 |
|
numpy_tile_op_test.py
|
|
|
|
one_hot_ops_test.py
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
onnx_while_test.py
|
|
|
|
order_switch_test.py
|
|
|
|
pack_ops_test.py
|
Change PackSegments to ensure consistent behavior between CPU and GPU
|
2019-12-30 13:31:45 -08:00 |
|
pack_rnn_sequence_op_test.py
|
|
|
|
pad_test.py
|
|
|
|
partition_ops_test.py
|
|
|
|
percentile_op_test.py
|
|
|
|
piecewise_linear_transform_test.py
|
|
|
|
pooling_test.py
|
[ROCm][Caffe2] Enable MIOpen 3D Pooling (#38260)
|
2020-07-08 17:42:55 -07:00 |
|
prepend_dim_test.py
|
|
|
|
python_op_test.py
|
|
|
|
quantile_test.py
|
[Rowwise Pruning][c2 op] Add Quantile Op (#32448)
|
2020-01-22 16:59:56 -08:00 |
|
rand_quantization_op_speed_test.py
|
|
|
|
rand_quantization_op_test.py
|
Disables flaky test_rand_quantization (#29463)
|
2019-11-08 13:51:22 -08:00 |
|
rank_loss_operator_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
rebatching_queue_test.py
|
|
|
|
record_queue_test.py
|
|
|
|
recurrent_net_executor_test.py
|
caffe2 - set up correct inheritance structure for remaining operator test classes (#18622)
|
2019-04-01 15:53:22 -07:00 |
|
recurrent_network_test.py
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
reduce_ops_test.py
|
|
|
|
reduction_ops_test.py
|
|
|
|
reshape_ops_test.py
|
ReshapeOp supports empty tensor (#21230)
|
2019-06-06 15:02:11 -07:00 |
|
resize_op_test.py
|
|
|
|
rmac_regions_op_test.py
|
|
|
|
rnn_cell_test.py
|
|
|
|
roi_align_rotated_op_test.py
|
|
|
|
rowwise_counter_test.py
|
separate op for rowwise counter (#31612)
|
2019-12-27 12:18:39 -08:00 |
|
scale_op_test.py
|
ScaleBlobs Operator (#19660)
|
2019-05-08 17:57:33 -07:00 |
|
segment_ops_test.py
|
|
|
|
self_binning_histogram_test.py
|
Back out "Revert D22329069: Self binning histogram" (#41313)
|
2020-07-13 11:46:34 -07:00 |
|
selu_op_test.py
|
|
|
|
sequence_ops_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
shape_inference_test.py
|
add conversion functions to embedding tables (#31083)
|
2020-01-08 16:56:12 -08:00 |
|
sinusoid_position_encoding_op_test.py
|
|
|
|
softmax_ops_test.py
|
Support softmax with D == 0 (#29167)
|
2019-11-11 00:46:10 -08:00 |
|
softplus_op_test.py
|
|
|
|
sparse_dropout_with_replacement_op_test.py
|
Implement dropout with replacement for id list features. (#22880)
|
2019-07-23 14:34:21 -07:00 |
|
sparse_gradient_checker_test.py
|
|
|
|
sparse_lengths_sum_benchmark.py
|
[caffe2] use 2-stage EmbeddingSpMDM interface (#32271)
|
2020-01-22 19:05:36 -08:00 |
|
sparse_lp_regularizer_test.py
|
Adding sparse Lp regularization operator to Caffe2 (#38574)
|
2020-06-01 15:21:19 -07:00 |
|
sparse_normalize_test.py
|
Make SparseNormalize backwards compatible (#25660)
|
2019-09-05 15:14:21 -07:00 |
|
sparse_ops_test.py
|
idtt + zch distributed inference (#35763)
|
2020-04-03 12:09:34 -07:00 |
|
sparse_to_dense_mask_op_test.py
|
|
|
|
spatial_bn_op_test.py
|
[Caffe2] Fix spatial_bn bug for computing running_var on CPU or on CUDA without CuDNN (#42151)
|
2020-07-29 11:20:03 -07:00 |
|
specialized_segment_ops_test.py
|
|
|
|
square_root_divide_op_test.py
|
|
|
|
stats_ops_test.py
|
|
|
|
stats_put_ops_test.py
|
|
|
|
storm_test.py
|
Implementation and operator test for STORM optimizer (#36225)
|
2020-04-14 23:04:26 -07:00 |
|
string_ops_test.py
|
|
|
|
text_file_reader_test.py
|
|
|
|
thresholded_relu_op_test.py
|
|
|
|
tile_op_test.py
|
|
|
|
top_k_test.py
|
|
|
|
torch_integration_test.py
|
Export logic op to pytorch
|
2020-07-08 02:27:09 -07:00 |
|
transpose_op_test.py
|
Fix bug in caffe2 transpose on GPU (#22233)
|
2019-06-26 11:33:25 -07:00 |
|
trigonometric_op_test.py
|
Update cafe2 hypothesis_test_util to support hypothesis-5 (#39498)
|
2020-06-05 08:27:50 -07:00 |
|
unique_ops_test.py
|
[ROCm] Skip caffe2 unique op test for rocm3.5 (#41219)
|
2020-07-09 20:00:29 -07:00 |
|
unique_uniform_fill_op_test.py
|
|
|
|
upsample_op_test.py
|
|
|
|
utility_ops_test.py
|
[Caffe2][Pruning] Make the caffe2 Sum operator support long types (#40379)
|
2020-06-23 14:18:29 -07:00 |
|
video_input_op_test.py
|
[caffe2] fix test error in video_input_op_test (#39382)
|
2020-06-02 11:48:01 -07:00 |
|
weight_scale_test.py
|
[c2] cuda impl for WeightScale op (#38712)
|
2020-05-26 12:50:54 -07:00 |
|
weighted_multi_sample_test.py
|
|
|
|
weighted_sample_test.py
|
|
|
|
weighted_sum_test.py
|
|
|
|
wngrad_test.py
|
remove unused parameters in optimizer tests (#18084)
|
2019-03-15 18:06:15 -07:00 |
