pytorch/caffe2/python/operator_test
Mayank Rana 1c414426df Caffe2: Cuda implementation for BatchOneHot operator
Summary: Cuda implementation for BatchOneHot operator.

Reviewed By: lvdmaaten

Differential Revision: D5639080

fbshipit-source-id: 8ee280c4bab64c1fdfb7429ee2c9ac8c02933931
2017-09-11 08:24:44 -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
apmeter_test.py
atomic_ops_test.py
batch_box_cox_test.py
blobs_queue_db_test.py Caffe2: support Tensor in BlobsQueueDB 2017-08-11 11:21:14 -07:00
boolean_mask_test.py fp16: SequenceMask 2017-09-09 13:02:38 -07:00
boolean_unmask_test.py Add CUDA implementation of BooleanUnmask and fixed some bugs in the test 2017-08-01 16:51:40 -07:00
channel_shuffle_test.py Opensourcing channel shuffle 2017-08-25 16:46:31 -07:00
checkpoint_test.py
clip_op_test.py Vectorize CPU ClipOp implementation (and add test) 2017-06-26 11:33:13 -07:00
concat_split_op_test.py Add tests for ConcatOp and SplitOp 2017-08-18 14:02:42 -07:00
conditional_test.py Use char-ngram embedding for out-of-vocabulary words 2017-09-01 19:16:49 -07:00
conv_test.py Adding 1d-2d-3d Schemas for Conv and Pool 2017-08-17 09:45:54 -07:00
conv_transpose_test.py Make ConvTranspose allow optional bias term 2017-09-07 17:16:43 -07:00
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 Fix cell/hidden init issue, add copy states to test 2017-09-06 14:16:17 -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 L1Distance Row-wise, instead of cumulative 2017-08-07 14:09:25 -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
elementwise_op_broadcast_test.py
elementwise_ops_test.py
emptysample_ops_test.py
extend_tensor_op_test.py
fc_operator_test.py fp16: add test for FC 2017-09-08 10:34:34 -07:00
filler_ops_test.py Caffe2: diagonal fill op 2017-08-16 13:05:11 -07:00
find_op_test.py
gather_ops_test.py GPU version of BatchGatherOp 2017-08-17 18:31:10 -07:00
gather_ranges_op_test.py
given_tensor_fill_op_test.py
group_conv_test.py
gru_test.py Revert D5589309: modify _LSTM into _RNN to adapt GRU 2017-08-10 16:42:41 -07:00
hsm_test.py Core unit test fixes for Python 3 2017-06-23 13:22:16 -07:00
im2col_col2im_test.py Implement gradients for Col2Im and Im2Col operators 2017-08-07 15:51:30 -07:00
image_input_op_test.py ImageInputOp_more_data_augmentation 2017-08-19 14:15:58 -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
layer_norm_op_test.py Layer norm brew wrapper 2017-08-17 11:17:47 -07:00
leaky_relu_test.py minor leaky relu bug fixes 2017-07-10 16:04:45 -07:00
lengths_tile_op_test.py
lengths_top_k_ops_test.py implement LengthsTopK operator 2017-08-07 18:19:29 -07:00
loss_ops_test.py
lpnorm_op_test.py adding operator lp_norm to support calculating l1 norm and l2 norm 2017-08-02 15:09:08 -07:00
map_ops_test.py CreateMapOp 2017-08-09 13:32:19 -07:00
margin_ranking_criterion_op_test.py
matmul_op_test.py
merge_id_lists_op_test.py Operator to Merge ID_LIST features 2017-08-17 01:16:00 -07:00
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 axis argument to NormalizeOp and NormalizeGradientOp 2017-09-05 11:17:32 -07:00
one_hot_ops_test.py Caffe2: Cuda implementation for BatchOneHot operator 2017-09-11 08:24:44 -07:00
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 Added fast path for CUDNN global max pooling 2017-08-23 16:33:06 -07:00
pow_op_test.py
prepend_dim_test.py PrependDimOp 2017-08-24 18:52:05 -07:00
python_op_test.py
rank_loss_operator_test.py improve pair_wise_loss operator to support multiple sessions 2017-07-28 15:12:47 -07:00
rebatching_queue_test.py
record_queue_test.py
recurrent_net_executor_test.py threaded RNN executor for CPU, multi-stream executor CUDA 2017-09-06 12:26:30 -07:00
recurrent_network_test.py threaded RNN executor for CPU, multi-stream executor CUDA 2017-09-06 12:26:30 -07:00
reduce_ops_test.py
reduction_ops_test.py fp16: SumSqrElements 2017-09-08 16:36:51 -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 soft-coverage attention 2017-08-31 21:21:54 -07:00
segment_ops_test.py CUDA SparseLengthsWeightedSum 2017-08-22 15:42:02 -07:00
sequence_ops_test.py
shape_inference_test.py Fix shape inference of distance_op 2017-09-07 17:16:46 -07:00
sinusoid_position_encoding_op_test.py Optimizations for Caffe2 SinusoidPositionEncodingOp 2017-08-22 00:04:06 -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
sparse_to_dense_mask_op_test.py add gradient for SparseToDenseMask operator 2017-08-01 13:05:03 -07:00
spatial_bn_op_test.py Allow inplace for spatial_bn_op 2017-08-17 09:18:55 -07:00
specialized_segment_ops_test.py Code generator for and high-performance emebding look-up kernels, supporting 2017-08-30 16:22:11 -07:00
square_root_divide_op_test.py float support for square root divide 2017-07-27 17:40:40 -07:00
stats_ops_test.py Core unit test fixes for Python 3 2017-06-23 13:22:16 -07:00
string_ops_test.py
text_file_reader_test.py Core unit test fixes for Python 3 2017-06-23 13:22:16 -07:00
tile_op_test.py
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 Adding a range operator similar to np.arange 2017-08-18 14:45:56 -07:00
video_input_op_test.py