Cheng Cheng fd18b89c98 shape inference for learning rate op (#20020) ...

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

Add shape inference for LearningRate op. The output (lr) should have similar shape with input (iteration), but not the same type (float vs int).

Reviewed By: un-disclosed

Differential Revision: D15112300

fbshipit-source-id: 09969aefa15172a6f3c70cd9b2548e3020da5d7a

2019-05-14 23:34:32 -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

remove unused parameters in optimizer tests (#18084)

2019-03-15 18:06:15 -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

…

affine_channel_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

apmeter_test.py

…

arg_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

assert_test.py

…

atomic_ops_test.py

…

basic_rnn_test.py

…

batch_box_cox_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

batch_bucketize_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

batch_moments_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

batch_sparse_to_dense_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

bbox_transform_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

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

Enable boolean_mask, adadelta, adagrad fp16 on ROCm (#17235)

2019-02-27 10:07:36 -08:00

boolean_unmask_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

box_with_nms_limit_op_test.py

support different class modes for bbox in box_with_nms_limit_op

2019-04-30 16:02:44 -07:00

cast_op_test.py

support casting to string (#15110)

2018-12-12 21:33:58 -08:00

ceil_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

channel_backprop_stats_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

channel_shuffle_test.py

add NCHW2NHWC and NHWC2NCHW in utils.py (#15588)

2018-12-28 17:34:50 -08:00

channel_stats_op_test.py

Optimize channel_stats_op (#16243)

2019-03-12 12:08:00 -07:00

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

Fix clip gradient with empty input (#14709)

2018-12-05 22:53:25 -08:00

clip_tensor_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

collect_and_distribute_fpn_rpn_proposals_op_test.py

Enable CircleCI for Linux jobs (#12389)

2018-10-08 17:09:37 -07:00

concat_split_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

conditional_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

conftest.py

operator serialized test coverage summary document (#13703)

2018-11-09 15:04:08 -08:00

conv_test.py

Unify gpu_support variable in python tests (#16748)

2019-02-07 00:29:51 -08:00

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

cosine_embedding_criterion_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

counter_ops_test.py

…

crf_test.py

CircleCI: enable all flaky tests (#13356)

2018-10-31 09:34:49 -07:00

cross_entropy_ops_test.py

Enable CircleCI for Linux jobs (#12389)

2018-10-08 17:09:37 -07:00

ctc_beam_search_decoder_op_test.py

caffe2 - support flaky operator tests for caffe2 build (#18155)

2019-03-25 16:58:34 -07:00

ctc_greedy_decoder_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

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

No Op Optimizer (#12390)

2018-10-10 18:09:46 -07:00

dataset_ops_test.py

typo fix

2019-03-06 11:36:44 -08: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

add dense vector to id_list operator (#15090)

2018-12-18 16:27:38 -08:00

depthwise_3x3_conv_test.py

add NCHW2NHWC and NHWC2NCHW in utils.py (#15588)

2018-12-28 17:34:50 -08:00

detectron_keypoints.py

…

distance_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

dropout_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

duplicate_operands_test.py

…

elementwise_linear_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

elementwise_logical_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

elementwise_op_broadcast_test.py

Unify gpu_support variable in python tests (#16748)

2019-02-07 00:29:51 -08:00

elementwise_ops_test.py

Enable CircleCI for Linux jobs (#12389)

2018-10-08 17:09:37 -07:00

emptysample_ops_test.py

…

enforce_finite_op_test.py

…

ensure_clipped_test.py

…

ensure_cpu_output_op_test.py

…

erf_op_test.py

Export PyTorch erf to ONNX Erf and add Caffe2 Erf operator

2019-01-17 09:18:08 -08:00

expand_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

fc_operator_test.py

Integrate rocBLAS fp16 api into Caffe2 (#14882)

2018-12-10 17:54:06 -08:00

feature_maps_ops_test.py

…

filler_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

find_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

flatten_op_test.py

…

flexible_top_k_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

floor_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

gather_ops_test.py

Implements Gather operator for arbitrary axis, sharing the code with BatchGather. (#13756)

2018-12-04 11:54:28 -08:00

gather_ranges_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

given_tensor_byte_string_to_uint8_fill_op_test.py

…

given_tensor_fill_op_test.py

…

glu_op_test.py

Fix test_glu_old HealthCheck with smarter generation strategy. (#12975)

2018-10-24 13:45:19 -07:00

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

Enable CircleCI for Linux jobs (#12389)

2018-10-08 17:09:37 -07:00

heatmap_max_keypoint_op_test.py

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

2019-04-01 15:53:22 -07:00

hsm_test.py

…

hyperbolic_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

im2col_col2im_test.py

CircleCI: enable all flaky tests (#13356)

2018-10-31 09:34:49 -07:00

image_input_op_test.py

…

index_hash_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

index_ops_test.py

…

instance_norm_test.py

add NCHW2NHWC and NHWC2NCHW in utils.py (#15588)

2018-12-28 17:34:50 -08:00

integral_image_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

jsd_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

key_split_ops_test.py

…

lars_test.py

…

layer_norm_op_test.py

Add elementwise_affine for LayerNormGradientOp (#19982)

2019-05-03 15:33:46 -07:00

leaky_relu_test.py

add NCHW2NHWC and NHWC2NCHW in utils.py (#15588)

2018-12-28 17:34:50 -08:00

learning_rate_adaption_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

learning_rate_op_test.py

Add Gate Policy GateLearningRateOp (#20044)

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

length_split_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

lengths_pad_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

lengths_tile_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

lengths_top_k_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

listwise_l2r_operator_test.py

add non expotential emphasis loss to Lambdarank

2018-11-13 14:54:04 -08: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

add NCHW2NHWC and NHWC2NCHW in utils.py (#15588)

2018-12-28 17:34:50 -08:00

loss_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

lpnorm_op_test.py

…

map_ops_test.py

…

margin_ranking_criterion_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

math_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

matmul_op_test.py

Integrate rocBLAS fp16 api into Caffe2 (#14882)

2018-12-10 17:54:06 -08:00

mean_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

merge_id_lists_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

mkl_conv_op_test.py

…

mkl_packed_fc_op_test.py

…

mkl_speed_test.py

…

mod_op_test.py

…

moments_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

momentum_sgd_test.py

Unify gpu_support variable in python tests (#16748)

2019-02-07 00:29:51 -08:00

mpi_test.py

…

negate_gradient_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

ngram_ops_test.py

…

normalize_op_test.py

…

numpy_tile_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

one_hot_ops_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

onnx_while_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

order_switch_test.py

add NCHW2NHWC and NHWC2NCHW in utils.py (#15588)

2018-12-28 17:34:50 -08:00

pack_ops_test.py

cuda implementation for PackSegment to support presence mask (#14635)

2018-11-30 16:54:10 -08:00

pack_rnn_sequence_op_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

pad_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

partition_ops_test.py

…

percentile_op_test.py

…

piecewise_linear_transform_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

pooling_test.py

Unify gpu_support variable in python tests (#16748)

2019-02-07 00:29:51 -08:00

prepend_dim_test.py

Unify cuda and hip device types in Caffe2 python front end (#14221)

2018-11-29 14:00:16 -08:00

python_op_test.py

Remove numba dependency

2018-10-23 17:03:47 -07:00

rand_quantization_op_speed_test.py

…

rand_quantization_op_test.py

Enable CircleCI for Linux jobs (#12389)

2018-10-08 17:09:37 -07:00

rank_loss_operator_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -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

CircleCI: enable all flaky tests (#13356)

2018-10-31 09:34:49 -07:00

reduce_ops_test.py

CircleCI: enable all flaky tests (#13356)

2018-10-31 09:34:49 -07:00

reduction_ops_test.py

Enable more caffe2 fp16 rocm tests (#15040)

2018-12-10 21:30:21 -08:00

reshape_ops_test.py

Caffe 2: Reshape Op upgrade (#15380)

2019-01-13 22:49:40 -08:00

resize_op_test.py

Add NHWC support to Resize Operator (#15553)

2019-01-08 16:44:17 -08:00

rmac_regions_op_test.py

…

rnn_cell_test.py

Enable more caffe2 fp16 rocm tests (#15040)

2018-12-10 21:30:21 -08:00

roi_align_rotated_op_test.py

Unify cuda and hip device types in Caffe2 python front end (#14221)

2018-11-29 14:00:16 -08:00

scale_op_test.py

ScaleBlobs Operator (#19660)

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

segment_ops_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

selu_op_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

sequence_ops_test.py

Enable CircleCI for Linux jobs (#12389)

2018-10-08 17:09:37 -07:00

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

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

softmax_ops_test.py

Unify gpu_support variable in python tests (#16748)

2019-02-07 00:29:51 -08:00

softplus_op_test.py

…

sparse_gradient_checker_test.py

…

sparse_lengths_sum_benchmark.py

…

sparse_normalize_test.py

…

sparse_ops_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

sparse_to_dense_mask_op_test.py

…

spatial_bn_op_test.py

Unify gpu_support variable in python tests (#16748)

2019-02-07 00:29:51 -08:00

specialized_segment_ops_test.py

handle empty inputs to SparseLengthsMean correctly (#15389)

2018-12-21 22:20:14 -08:00

square_root_divide_op_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

stats_ops_test.py

…

stats_put_ops_test.py

Fix overflow error in stats_put_ops

2018-11-06 15:41:51 -08:00

string_ops_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

text_file_reader_test.py

…

thresholded_relu_op_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

tile_op_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

top_k_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

torch_integration_test.py

support different class modes for bbox in box_with_nms_limit_op

2019-04-30 16:02:44 -07:00

transpose_op_test.py

Unify gpu_support variable in python tests (#16748)

2019-02-07 00:29:51 -08:00

trigonometric_op_test.py

Refactor tests part 2 (#11811)

2018-09-19 10:09:28 -07:00

unique_ops_test.py

Enable arg_ops_test/unique_ops_test on AMD/rocm (#16853)

2019-02-07 16:51:15 -08:00

unique_uniform_fill_op_test.py

…

upsample_op_test.py

caffe2: UpsampleBilinear CUDA implementation (#12843)

2018-10-25 11:10:04 -07:00

utility_ops_test.py

Clean up a couple of items in the C2 test scaffolding (WIP) (#7847)

2018-11-07 09:16:13 -08:00

video_input_op_test.py

…

weighted_multi_sample_test.py

…

weighted_sample_test.py

…

weighted_sum_test.py

Refactor tests part 1 (#11350)

2018-09-18 10:42:10 -07:00

wngrad_test.py

remove unused parameters in optimizer tests (#18084)

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