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

…

adam_test.py

…

atomic_ops_test.py

…

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

…

conv_test.py

ConvNd cuDNN

2017-05-19 15:20:33 -07:00

conv_transpose_test.py

…

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

…

counter_ops_test.py

…

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

…

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

…

elementwise_ops_test.py

Vectorize SigmoidOp on CPU

2017-05-17 20:33:36 -07:00

emptysample_ops_test.py

…

extend_tensor_op_test.py

…

fc_operator_test.py

…

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

…

gather_ranges_op_test.py

…

given_tensor_fill_op_test.py

…

group_conv_test.py

…

hsm_test.py

…

image_input_op_test.py

deprecate CNNModelHelper in python/operator_test dir

2017-05-19 12:17:17 -07:00

index_ops_test.py

…

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

…

margin_ranking_criterion_op_test.py

…

matmul_op_test.py

…

mkl_conv_op_test.py

…

mkl_packed_fc_op_test.py

…

mkl_speed_test.py

deprecate CNNModelHelper in python/operator_test dir

2017-05-19 12:17:17 -07:00

momentum_sgd_test.py

…

mpi_test.py

…

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

…

pad_test.py

…

partition_ops_test.py

…

piecewise_linear_transform_test.py

…

pooling_test.py

Generalize PoolingOp(cuDNN) to compute 2D and 3D pooling.

2017-05-19 10:19:00 -07:00

pow_op_test.py

…

python_op_test.py

…

rank_loss_operator_test.py

…

rebatching_queue_test.py

caffe2: rebatching queue for MultiTask

2017-05-02 15:22:46 -07:00

record_queue_test.py

…

recurrent_network_test.py

deprecate CNNModelHelper in python/operator_test dir

2017-05-19 12:17:17 -07:00

reduce_ops_test.py

…

reduction_ops_test.py

…

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

…

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

…

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

…

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

…

stats_ops_test.py

…

string_ops_test.py

…

text_file_reader_test.py

…

tile_op_test.py

fix tileop test

2017-05-24 18:32:01 -07:00

top_k_test.py

…

unique_uniform_fill_op_test.py

…

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