Geet Sethi 2dc8851206 RNN Workspace Blob Extraction ...

Summary:
Added operator RecurrentNetworkBlobFetcherOp that takes as input a scratch workspace name and prefix, and copies over all blobs in the scratch workspace into the global workspace. This essentially extracts all intermediate recurrent network computation for each timestep.

Added a wrapper in recurrent.py - retrieve_step_blobs(net, prefix='rnn') - which, when called after an rnn is run, will return a list of all blobs extracted from the net.

Reviewed By: akyrola

Differential Revision: D5421926

fbshipit-source-id: 0f35b466d77d3c719fb0e32de7dbcafc6c0d5225

2017-07-17 10:24:18 -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

…

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

…

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

…

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

…

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

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

…

shape_inference_test.py

…

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_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