pytorch/caffe2/python/operator_test
Ahmed Taei 94d42b03fb MaxReduction ops GPU implementation.
Summary:
Move rowwise-max kernel from Softmax to math_util library and implement
colwwise-max kernel and MaxReduction ops.

Reviewed By: akyrola

Differential Revision: D5240329

fbshipit-source-id: a07281a877324de459aace33ff21175a68cfd8f6
2017-06-14 11:02:46 -07:00
..
activation_ops_test.py
adagrad_test.py Fix SparseAdagrad for indices.ndim>1 2017-05-30 12:02:18 -07:00
adam_test.py Fix SparseAdagrad for indices.ndim>1 2017-05-30 12:02:18 -07:00
apmeter_test.py Fixing broken Python tests 2017-06-08 13:34:46 -07:00
atomic_ops_test.py
batch_box_cox_test.py
boolean_unmask_test.py Fixes range/xrange for Python 3 2017-06-07 00:04:26 -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
cudnn_recurrent_test.py deprecate CNNModelHelper in python/operator_test dir 2017-05-19 12:17:17 -07:00
dataset_ops_test.py Fixes range/xrange for Python 3 2017-06-07 00:04:26 -07:00
distance_op_test.py Implemented L1Distance Operator for CUDA 2017-05-17 17:32:53 -07:00
dropout_op_test.py Mark in-place GPU dropout as broken, add test 2017-05-30 12:35:22 -07:00
duplicate_operands_test.py
elementwise_linear_op_test.py
elementwise_logical_ops_test.py Fixing some tests 2017-05-30 09:16:48 -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 Make fills work with input_shape when run in CUDAContext 2017-05-30 20:47:16 -07:00
find_op_test.py
gather_ops_test.py
gather_ranges_op_test.py
given_tensor_fill_op_test.py
group_conv_test.py
hsm_test.py Fixes range/xrange for Python 3 2017-06-07 00:04:26 -07:00
image_input_op_test.py Skip Python tests which require opencv or lmdb 2017-06-08 13:34:43 -07:00
index_ops_test.py Fixes range/xrange for Python 3 2017-06-07 00:04:26 -07:00
instance_norm_test.py Fixes zip/izip for Python 3 2017-06-07 00:04:26 -07:00
leaky_relu_test.py Fixes zip/izip for Python 3 2017-06-07 00:04:26 -07:00
lengths_tile_op_test.py
loss_ops_test.py
map_ops_test.py map operator (move maptrait def out of class) 2017-06-12 14:52:04 -07:00
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
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 Remove map() and filter() in favor of comprehensions 2017-05-30 15:32:58 -07:00
reduce_ops_test.py
reduction_ops_test.py MaxReduction ops GPU implementation. 2017-06-14 11:02:46 -07:00
relu_op_test.py
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 Static RNN 2017-06-08 17:48:48 -07:00
segment_ops_test.py Test smaller tensors in segment_ops_test 2017-05-31 20:17:31 -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 Fixes range/xrange for Python 3 2017-06-07 00:04:26 -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 Remove map() and filter() in favor of comprehensions 2017-05-30 15:32:58 -07:00
text_file_reader_test.py Fixes zip/izip for Python 3 2017-06-07 00:04:26 -07:00
tile_op_test.py Avoid numpy VisibleDeprecationWarning in test 2017-05-31 20:01:30 -07:00
top_k_test.py Fixing broken Python tests 2017-06-08 13:34:46 -07:00
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 Skip Python tests which require opencv or lmdb 2017-06-08 13:34:43 -07:00