OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ec19b4bd7b
pytorch/caffe2/python/operator_test
History
Anmol Kalia 7f98dc28cb Refactored spatial softmax 
Summary: Refactored SoftmaxWithLoss by removing the code for spatial=1 mode and created a new op SpatialSoftmaxWithLoss that has the spatial mode implemented.

Reviewed By: viswanathgs

Differential Revision: D5104120

fbshipit-source-id: 8ab999e32c916b2a39a670a7b2a3365401535f24
2017-05-26 14:50:43 -07:00
..
activation_ops_test.py CUDA version of PRelu/Gradient + Fix Gradient for dW 2017-04-24 15:52:25 -07:00
adagrad_test.py Add tests and GPU impls for sparse optimizers 2017-04-13 11:07:40 -07:00
adam_test.py Add tests and GPU impls for sparse optimizers 2017-04-13 11:07:40 -07:00
atomic_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
batch_box_cox_test.py add box cox transform op 2017-04-27 22:06:43 -07:00
boolean_unmask_test.py Caffe2 unit test for unmask 2017-04-18 21:06:18 -07:00
checkpoint_test.py snapshot -> checkpoint 2016-12-15 12:01:30 -08:00
conv_test.py ConvNd cuDNN 2017-05-19 15:20:33 -07:00
conv_transpose_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
copy_ops_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
cosine_embedding_criterion_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
counter_ops_test.py AtomicCounter to return previous value on Reset. 2017-02-02 14:59:30 -08:00
crf_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
cross_entropy_ops_test.py CUDA version of SigmoidCrossEntropyWithLogits 2017-04-14 16:07:33 -07:00
cudnn_recurrent_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
dataset_ops_test.py Generalize LastNWindowCollector 2017-05-04 16:05:15 -07:00
distance_op_test.py Implemented L1Distance Operator for CUDA 2017-05-17 17:32:53 -07:00
duplicate_operands_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
elementwise_linear_op_test.py ElementwiseLinearOp 2017-04-17 14:18:27 -07:00
elementwise_logical_ops_test.py Add row-wise broadcasting to "Where" operator 2017-04-27 12:31:54 -07:00
elementwise_op_broadcast_test.py new SumReduceLike op CPU/GPU implementation and doc 2017-04-13 10:28:46 -07:00
elementwise_ops_test.py Vectorize SigmoidOp on CPU 2017-05-17 20:33:36 -07:00
emptysample_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
extend_tensor_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
fc_operator_test.py Test for FC operator + fix for docs 2017-01-27 10:44:24 -08:00
filler_ops_test.py fix unit test 2017-04-19 17:22:00 -07:00
find_op_test.py FindOp 2017-04-24 15:52:25 -07:00
gather_ops_test.py Add GatherOp for GPU, and update its tests. 2017-03-31 13:20:09 -07:00
gather_ranges_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
given_tensor_fill_op_test.py Fixes for ops without a CUDA backend 2017-03-29 14:36:09 -07:00
group_conv_test.py Make all convolution operators allow optional bias term 2016-12-21 15:14:24 -08:00
hsm_test.py Generate huffman tree 2017-01-19 16:14:23 -08:00
image_input_op_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
index_ops_test.py Change the schema of IndexLoad & IndexFreeze so that state change is captured by the framework 2017-02-14 10:05:12 -08:00
instance_norm_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
leaky_relu_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
lengths_tile_op_test.py add gradient for LengthsTileOp 2017-04-25 14:31:15 -07:00
loss_ops_test.py Caffe2: consolidate AveragedLoss with SumElementsOp 2017-04-06 10:35:01 -07:00
margin_ranking_criterion_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
matmul_op_test.py BatchMatMulOp: use cuBLAS batched strided gemm for CUDA 2017-03-28 11:54:09 -07:00
mkl_conv_op_test.py MKL convolution operator 2017-01-23 09:59:30 -08:00
mkl_packed_fc_op_test.py MKL convolution operator 2017-01-23 09:59:30 -08:00
mkl_speed_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
momentum_sgd_test.py Add tests and GPU impls for sparse optimizers 2017-04-13 11:07:40 -07:00
mpi_test.py Setup MPI before test start 2016-12-19 15:59:32 -08:00
normalize_op_test.py NormalizeOP CUDA impelementation 2017-05-01 09:25:30 -07:00
one_hot_ops_test.py Fix tests for ops without a CUDA backend 2017-04-24 15:52:25 -07:00
pack_ops_test.py Registering GPU version of PackSegments using GPUFallbackOp 2017-03-24 16:01:53 -07:00
pad_test.py Support cropping with negative pad sizes in PadImage 2017-04-03 23:47:54 -07:00
partition_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
piecewise_linear_transform_test.py PiecewiseLinearTransformOp supports passing params from input blobs. 2017-04-08 11:02:35 -07:00
pooling_test.py Generalize PoolingOp(cuDNN) to compute 2D and 3D pooling. 2017-05-19 10:19:00 -07:00
pow_op_test.py CUDA version of elementwise power + rename to Pow + gradient 2017-03-07 10:20:40 -08:00
python_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
rank_loss_operator_test.py Normalize rank loss gradient to avoid convergence issues when the number of pairs is really large 2016-12-21 17:29:24 -08:00
rebatching_queue_test.py caffe2: rebatching queue for MultiTask 2017-05-02 15:22:46 -07:00
record_queue_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
recurrent_network_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
reduce_ops_test.py ReduceBack{Sum|Mean}Op CPU & GPU implementation 2017-03-13 16:19:58 -07:00
reduction_ops_test.py SumSqrElements 2017-04-10 16:16:52 -07:00
relu_op_test.py Change back the function signature of relu gradient to only use 2017-04-13 22:08:09 -07:00
reshape_ops_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
resize_op_test.py Add specialized ResizeNearest implementation for scale=2 2017-05-07 21:10:11 -07:00
rnn_cell_test.py gradient accumulation fix 2017-05-24 10:33:32 -07:00
segment_ops_test.py UnsortedSegmentSum/Mean for CUDA 2017-05-26 09:33:49 -07:00
sequence_ops_test.py add gpu support for caffe2-seq2seq 2017-03-17 05:19:14 -07:00
shape_inference_test.py Refactored spatial softmax 2017-05-26 14:50:43 -07:00
softmax_ops_test.py Refactored spatial softmax 2017-05-26 14:50:43 -07:00
softplus_op_test.py softplus op 2017-05-08 10:40:25 -07:00
sparse_gradient_checker_test.py Fixes for ops without a CUDA backend 2017-03-29 14:36:09 -07:00
sparse_ops_test.py Add ScatterWeightedSum for GPU. 2017-05-03 13:40:48 -07:00
spatial_bn_op_test.py Support 3D&1D SpatialBatchNorm[CPU] 2017-05-11 09:32:54 -07:00
square_root_divide_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
stats_ops_test.py Performance counters 2017-02-21 16:31:24 -08:00
string_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
text_file_reader_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
tile_op_test.py fix tileop test 2017-05-24 18:32:01 -07:00
top_k_test.py Implement TopK op in caffe2 2017-03-16 17:32:20 -07:00
unique_uniform_fill_op_test.py UniqueUniformFillOp 2017-02-15 16:00:44 -08:00
utility_ops_test.py Implement SizeOp as requested in github issue#583 2017-05-25 11:07:35 -07:00
video_input_op_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00