pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

Yanghan Wang e34ef04301 register HeatmapMaxKeypoint with C10 (#25191 )

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25191

registering as C10.

Test Plan: buck test mode/dev-nosan caffe2/caffe2/python/operator_test:heatmap_max_keypoint_op_test

Reviewed By: newstzpz

Differential Revision: D17056321

fbshipit-source-id: 989b72d7e3c9f23684b10d5fc9b98177ad4ee47b

2019-08-27 20:13:57 -07:00

__init__.py

…

activation_ops_test.py

Fix relu bug for empty tensor (#19451 )

2019-04-19 15:21:07 -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_helper.py

add decay parameter in ref_adagrad (#15329 )

2019-05-07 18:58:58 -07:00

adagrad_test.py

remove unused parameters in optimizer tests (#18084 )

2019-03-15 18:06:15 -07:00

adam_test.py

eliminate FE_INVALID in optimizer related operators and tests (#20501 )

2019-05-16 08:23:46 -07:00

affine_channel_op_test.py

…

apmeter_test.py

…

arg_ops_test.py

…

assert_test.py

…

atomic_ops_test.py

…

basic_rnn_test.py

…

batch_box_cox_test.py

…

batch_bucketize_op_test.py

…

batch_moments_op_test.py

…

batch_sparse_to_dense_op_test.py

…

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

Adding gradient to Boolean Mask operator (#21423 )

2019-06-06 20:48:47 -07:00

boolean_unmask_test.py

…

box_with_nms_limit_op_test.py

fix bug of not using get_score_cls_index in BoxWithNMSLimitOp (#20868 )

2019-05-24 22:31:01 -07:00

bucketize_op_test.py

Move bucketize_op to open source

2019-05-20 18:03:27 -07:00

cast_op_test.py

…

ceil_op_test.py

…

channel_backprop_stats_op_test.py

…

channel_shuffle_test.py

…

channel_stats_op_test.py

…

checkpoint_test.py

caffe2 - set up correct inheritance structure for remaining operator test classes (#18622 )

2019-04-01 15:53:22 -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

…

conditional_test.py

…

conftest.py

…

conv_test.py

…

conv_transpose_test.py

Add support for group ConvTranspose (#18794 )

2019-04-04 11:52:06 -07: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

…

cross_entropy_ops_test.py

…

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

…

data_couple_op_test.py

…

dataset_ops_test.py

…

deform_conv_test.py

…

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

…

elementwise_ops_test.py

…

emptysample_ops_test.py

…

enforce_finite_op_test.py

…

ensure_clipped_test.py

…

ensure_cpu_output_op_test.py

…

erf_op_test.py

…

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

…

filler_ops_test.py

…

find_op_test.py

…

flatten_op_test.py

…

flexible_top_k_test.py

…

floor_op_test.py

…

gather_ops_test.py

support Gather different indices for different examples in one batch (#23813 )

2019-08-07 21:14:30 -07:00

gather_ranges_op_test.py

…

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

…

group_norm_op_test.py

…

gru_test.py

…

heatmap_max_keypoint_op_test.py

2019-08-27 20:13:57 -07:00

hsm_test.py

…

hyperbolic_ops_test.py

…

im2col_col2im_test.py

…

image_input_op_test.py

…

index_hash_ops_test.py

…

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

Add Gate Policy GateLearningRateOp (#20044 )

2019-05-05 20:11:04 -07:00

length_split_op_test.py

…

lengths_pad_op_test.py

…

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

…

map_ops_test.py

…

margin_ranking_criterion_op_test.py

…

math_ops_test.py

eliminate FE_INVALID in unit test (#20502 )

2019-05-16 21:55:28 -07:00

matmul_op_test.py

…

mean_op_test.py

…

merge_id_lists_op_test.py

…

mkl_conv_op_test.py

…

mkl_packed_fc_op_test.py

…

mkl_speed_test.py

…

mod_op_test.py

…

moments_op_test.py

…

momentum_sgd_test.py

…

mpi_test.py

…

negate_gradient_op_test.py

…

ngram_ops_test.py

…

normalize_op_test.py

…

numpy_tile_op_test.py

…

one_hot_ops_test.py

…

onnx_while_test.py

…

order_switch_test.py

…

pack_ops_test.py

…

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

…

prepend_dim_test.py

…

python_op_test.py

…

rand_quantization_op_speed_test.py

…

rand_quantization_op_test.py

…

rank_loss_operator_test.py

…

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

…

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

…

scale_op_test.py

ScaleBlobs Operator (#19660 )

2019-05-08 17:57:33 -07:00

segment_ops_test.py

…

selu_op_test.py

…

sequence_ops_test.py

…

shape_inference_test.py

shape inference for learning rate op (#20020 )

2019-05-14 23:34:32 -07:00

sinusoid_position_encoding_op_test.py

…

softmax_ops_test.py

…

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

…

sparse_normalize_test.py

Remove gradient value as input from SparseNormalize op (#24357 )

2019-08-19 14:47:09 -07:00

sparse_ops_test.py

…

sparse_to_dense_mask_op_test.py

…

spatial_bn_op_test.py

…

specialized_segment_ops_test.py

…

square_root_divide_op_test.py

…

stats_ops_test.py

…

stats_put_ops_test.py

…

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

enable "keeps" from BoxWithNMSLimit and caffe2_fastrcnn_outputs_inference

2019-08-19 10:54:22 -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

…

unique_ops_test.py

…

unique_uniform_fill_op_test.py

…

upsample_op_test.py

…

utility_ops_test.py

…

video_input_op_test.py

update video input (#22471 )

2019-07-05 00:56:33 -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