pytorch/caffe2/python/operator_test
Bangsheng Tang d8443b8ffa BatchGatherOp
Summary:
1. added BatchGatherOp and BatchGatherGradientOp
2. unit tests

Reviewed By: xianjiec

Differential Revision: D5443965

fbshipit-source-id: bdcbb7f9f91c55484372a4bdb1727ae6d49e2018
2017-07-27 10:17:42 -07:00
..
activation_ops_test.py Fix flaky test due to numerical gradient approximation error. 2017-07-26 18:58:19 -07:00
adagrad_test.py Fix GPU SparseAdaGrad with empty tensors 2017-07-25 23:50:54 -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_mask_test.py Fix some bugs in CPU version of BooleanMask and add GPU version 2017-07-21 11:38:49 -07:00
boolean_unmask_test.py Fixes range/xrange for Python 3 2017-06-07 00:04:26 -07:00
checkpoint_test.py
clip_op_test.py Vectorize CPU ClipOp implementation (and add test) 2017-06-26 11:33:13 -07:00
conv_test.py Tests: fix test_convolution_sync() 2017-06-27 18:07:04 -07:00
conv_transpose_test.py
copy_ops_test.py
cosine_embedding_criterion_op_test.py
counter_ops_test.py
crf_test.py
cross_entropy_ops_test.py
cudnn_recurrent_test.py Dict fixes/improvements and unittest targets for Python 3 in caffe2 core 2017-06-29 17:05:41 -07:00
dataset_ops_test.py Core unit test fixes for Python 3 2017-06-23 13:22:16 -07:00
distance_op_test.py distance_op_test from hypothesis_test refactored 2017-07-26 13:37:08 -07:00
dropout_op_test.py Fixes for CuDNNDropoutOp 2017-06-15 22:51:12 -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
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 BatchGatherOp 2017-07-27 10:17:42 -07:00
gather_ranges_op_test.py
given_tensor_fill_op_test.py
group_conv_test.py
gru_test.py Implemented GRUCell 2017-07-10 17:52:25 -07:00
hsm_test.py Core unit test fixes for Python 3 2017-06-23 13:22:16 -07:00
image_input_op_test.py Tests: handle missing python-lmdb gracefully 2017-06-27 09:47:46 -07:00
index_hash_ops_test.py IndexHash 2017-07-07 23:06:11 -07:00
index_ops_test.py Core unit test fixes for Python 3 2017-06-23 13:22:16 -07:00
instance_norm_test.py Fixes zip/izip for Python 3 2017-06-07 00:04:26 -07:00
leaky_relu_test.py minor leaky relu bug fixes 2017-07-10 16:04:45 -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
momentum_sgd_test.py
mpi_test.py
normalize_op_test.py Add NormalizeL1Op for doing L1 nomalization along given axis 2017-07-10 10:10:36 -07:00
one_hot_ops_test.py
pack_ops_test.py
pack_rnn_sequence_op_test.py added PackRNNSequence and UnpackRNNSequence operators 2017-06-30 09:53:31 -07:00
pad_test.py
partition_ops_test.py
piecewise_linear_transform_test.py Added PiecewiseLinearTransform CUDA Op 2017-07-07 15:20:00 -07:00
pooling_test.py
pow_op_test.py
python_op_test.py
rank_loss_operator_test.py Fix test_pair_wise_loss_predictions 2017-06-29 12:48:04 -07:00
rebatching_queue_test.py
record_queue_test.py
recurrent_network_test.py RNN Workspace Blob Extraction 2017-07-17 10:24:18 -07:00
reduce_ops_test.py
reduction_ops_test.py Optimize SumSqrElementsOp for CUDA 2017-06-16 17:03:38 -07:00
relu_op_test.py
reshape_ops_test.py
resize_op_test.py added gradients for ResizeNearest (CPU + CUDA) and ref 2017-07-07 14:19:42 -07:00
rnn_cell_test.py use model.create_param instead of using param_init_net directly 2017-07-24 21:17:24 -07:00
segment_ops_test.py Add support for int32 indices in SparseLengthSum and friends 2017-07-24 17:50:00 -07:00
sequence_ops_test.py
shape_inference_test.py Fix Flatten operator for empty tensors 2017-07-25 17:51:42 -07:00
softmax_ops_test.py stabilize softmax_ops_test 2017-06-22 13:50:32 -07:00
softplus_op_test.py Fix gradient checking for softplus op 2017-06-21 05:35:43 -07:00
sparse_gradient_checker_test.py
sparse_lengths_sum_benchmark.py Add a quick SparseLengthsSum benchmark. 2017-07-24 13:17:47 -07:00
sparse_ops_test.py Fixes range/xrange for Python 3 2017-06-07 00:04:26 -07:00
spatial_bn_op_test.py
square_root_divide_op_test.py
stats_ops_test.py Core unit test fixes for Python 3 2017-06-23 13:22:16 -07:00
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 Core unit test fixes for Python 3 2017-06-23 13:22:16 -07:00
tile_op_test.py Fixes in tests to support numpy >= 0.12 2017-06-15 14:17:20 -07:00
top_k_test.py Implement TopKOp for GPU 2017-06-17 08:47:38 -07:00
unique_uniform_fill_op_test.py
utility_ops_test.py Allow specification of SliceOp dimensions via argument rather than via tensor 2017-06-27 17:17:32 -07:00
video_input_op_test.py Skip Python tests which require opencv or lmdb 2017-06-08 13:34:43 -07:00