pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00

Files

Robert Verkuil 97193478c7 Implemented GRUCell

Summary: Implemented python logic and tests to create an RNNCell for GRU.  Uses the preexisting GRU Unit Op code.

Reviewed By: salexspb

Differential Revision: D5364893

fbshipit-source-id: 2451d7ec8c2eacb8d8c9b7c893bfd21b65fb9d18

2017-07-10 17:52:25 -07:00

activation_ops_test.py

ELU CUDA implementation

2017-06-21 11:47:13 -07:00

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

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

Fixes range/xrange for Python 3

2017-06-07 00:04:26 -07:00

checkpoint_test.py

snapshot -> checkpoint

2016-12-15 12:01:30 -08:00

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

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

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

Implemented L1Distance Operator for CUDA

2017-05-17 17:32:53 -07:00

dropout_op_test.py

Fixes for CuDNNDropoutOp

2017-06-15 22:51:12 -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

Fixing some tests

2017-05-30 09:16:48 -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

Make fills work with input_shape when run in CUDAContext

2017-05-30 20:47:16 -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

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

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

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

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

Add NormalizeL1Op for doing L1 nomalization along given axis

2017-07-10 10:10:36 -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

pack_rnn_sequence_op_test.py

added PackRNNSequence and UnpackRNNSequence operators

2017-06-30 09:53:31 -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

Added PiecewiseLinearTransform CUDA Op

2017-07-07 15:20:00 -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

Fix test_pair_wise_loss_predictions

2017-06-29 12:48:04 -07: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

Fix a few typos and grammars in comment

2017-06-14 18:22:39 -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

Optimize SumSqrElementsOp for CUDA

2017-06-16 17:03:38 -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

added gradients for ResizeNearest (CPU + CUDA) and ref

2017-07-07 14:19:42 -07:00

rnn_cell_test.py

Moved sigmoid, tanh, and _prepare_lstm (renamed) to a util file.

2017-07-10 17:52:22 -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

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

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

Fixes for ops without a CUDA backend

2017-03-29 14:36:09 -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

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

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

UniqueUniformFillOp

2017-02-15 16:00:44 -08:00

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