pytorch/caffe2/python
Anders Papitto db6777eaf4 fix gru_cell bug
Summary:
the fc needs to be in the output_gate_t scope so it can find its input
weights correctly
Closes https://github.com/caffe2/caffe2/pull/1739

Reviewed By: dzhulgakov

Differential Revision: D6705443

Pulled By: anderspapitto

fbshipit-source-id: 139e83ac77589a203ffe404fedab98eea5b1a51c
2018-01-12 15:34:23 -08:00
..
docs
examples Allow shifting of activations / ops to other GPUs in data parallel model 2017-11-29 21:17:00 -08:00
helpers Add if and while ops to brew 2017-12-05 17:33:34 -08:00
layers add dense regularization 2018-01-08 13:03:17 -08:00
mint
mkl Add MKLConcatOp 2018-01-11 14:19:22 -08:00
modeling Added inverted FP16 Initializer 2017-10-27 10:20:04 -07:00
models Fix pool op custom path issue 2, wrongful routing to global pooling 2018-01-09 00:54:45 -08:00
operator_test Fix failed test due to D6665466 2018-01-09 16:40:35 -08:00
predictor Record workflow run id for inference. 2017-12-18 15:33:19 -08:00
rnn GRU cell: add linear_before_reset boolean parameter 2018-01-08 13:22:56 -08:00
test Async executor with less polling 2017-11-28 18:50:32 -08:00
_import_c_extension.py
allcompare_test.py
attention.py
benchmark_generator.py
binarysize.py
brew.py Add if and while ops to brew 2017-12-05 17:33:34 -08:00
brew_test.py Add if and while ops to brew 2017-12-05 17:33:34 -08:00
build.py
cached_reader.py Cached reader 2017-11-15 12:38:49 -08:00
caffe_translator.py
caffe_translator_test.py
checkpoint.py Generalize construction of db_name in checkpoint manager 2018-01-10 11:49:17 -08:00
checkpoint_test.py Generalize construction of db_name in checkpoint manager 2018-01-10 11:49:17 -08:00
CMakeLists.txt Fix OSS build 2017-12-21 19:04:25 -08:00
cnn.py
context.py
context_test.py
control.py
control_ops_grad.py Backpropagation for While op 2017-12-18 16:03:45 -08:00
control_ops_util.py Backpropagation for While op 2017-12-18 16:03:45 -08:00
control_test.py
convnet_benchmarks.py
convnet_benchmarks_test.py
core.py Backpropagation for While op 2017-12-18 16:03:45 -08:00
core_gradients_test.py
core_test.py Py3 test fixes 2017-12-05 10:34:41 -08:00
crf.py
data_parallel_model.py Allow shifting of activations / ops to other GPUs in data parallel model 2017-11-29 21:17:00 -08:00
data_parallel_model_test.py Skip DeviceShiftTest if host has < 4 GPU devices 2017-12-03 16:02:05 -08:00
data_parallel_model_utils.py Allow shifting of activations / ops to other GPUs in data parallel model 2017-11-29 21:17:00 -08:00
data_workers.py move print to logger 2017-11-17 18:03:44 -08:00
data_workers_test.py
dataio.py Adding a time limit reader 2018-01-02 11:33:53 -08:00
dataio_test.py Adding a time limit reader 2018-01-02 11:33:53 -08:00
dataset.py
db_test.py
device_checker.py
dlpack.h Support for DLPack in Python op 2017-12-21 17:02:16 -08:00
dyndep.py
embedding_generation_benchmark.py
experiment_util.py
extension_loader.py
functional.py
functional_test.py
gradient_check_test.py Backpropagation for While op 2017-12-18 16:03:45 -08:00
gradient_checker.py
gru_cell.py fix gru_cell bug 2018-01-12 15:34:23 -08:00
hsm_util.py
hypothesis_test.py Add CUDA implementation for ReplaceNaNOp 2017-12-05 13:34:51 -08:00
hypothesis_test_util.py Ensure indices list in sparse optimizer tests is unique 2018-01-03 12:19:14 -08:00
layer_model_helper.py enable setting model initialization seed 2018-01-11 14:04:03 -08:00
layer_model_instantiator.py add dense regularization 2018-01-08 13:03:17 -08:00
layer_parameter_sharing_test.py Add shape checks and print more info in parameter sharing 2017-10-27 01:22:06 -07:00
layer_test_util.py
layers_test.py use numerically stable version of BatchLRLoss 2018-01-02 13:18:36 -08:00
lengths_reducer_rowwise_8bit_ops_test.py
lstm_benchmark.py
memonger.py
memonger_test.py
mkl_test_util.py
model_device_test.py
model_helper.py add sanity check to model_helper.TensorProtosDBInput 2017-11-21 10:28:25 -08:00
modifier_context.py
mpi_python.cc Upgrade to 2.2.1 2017-10-22 13:26:56 -07:00
muji.py
muji_test.py
net_builder.py Minor documentation fix in NetBuiler 2017-11-15 16:22:22 -08:00
net_builder_test.py
net_drawer.py
net_printer.py Unit test that compares net snippets after parallelization 2017-11-08 15:55:27 -08:00
net_printer_test.py
observer_test.py Attach observers to operators inside step net 2017-11-14 15:06:38 -08:00
optimizer.py Revert changes in blob name in optimizer 2017-12-04 19:32:45 -08:00
optimizer_context.py
optimizer_test.py Support RMSProp in Caffe2. 2017-11-08 16:43:18 -08:00
optimizer_test_util.py momentum sgd 2017-11-03 16:17:17 -07:00
parallel_workers.py
parallel_workers_test.py
parallelize_bmuf_distributed_test.py BMUF cpu support 2017-11-19 23:41:25 -08:00
pipeline.py
pipeline_test.py
predictor_constants.py
pybind_state.cc Support for DLPack in Python op 2017-12-21 17:02:16 -08:00
pybind_state.h Support for DLPack in Python op 2017-12-21 17:02:16 -08:00
pybind_state_dlpack.cc Support for DLPack in Python op 2017-12-21 17:02:16 -08:00
pybind_state_dlpack.h Support for DLPack in Python op 2017-12-21 17:02:16 -08:00
pybind_state_gpu.cc Support for DLPack in Python op 2017-12-21 17:02:16 -08:00
pybind_state_mkl.cc
python_op_test.py Throw Python exception from PythonOp instead of logging 2017-11-20 09:03:17 -08:00
queue_util.py
record_queue.py
recurrent.py Remove scoping assertion because it is not useful and causing errors 2017-12-11 18:03:45 -08:00
regularizer.py add dense regularization 2018-01-08 13:03:17 -08:00
regularizer_context.py
regularizer_test.py add dense regularization 2018-01-08 13:03:17 -08:00
rnn_cell.py GRU cell: add linear_before_reset boolean parameter 2018-01-08 13:22:56 -08:00
schema.py add struct get method 2017-12-19 12:35:56 -08:00
schema_test.py add struct get method 2017-12-19 12:35:56 -08:00
scope.py Add a EmptyDeviceScope (i.e. allow setting CurrentDeviceScope() to None) 2017-11-02 11:25:48 -07:00
scope_test.py Add a EmptyDeviceScope (i.e. allow setting CurrentDeviceScope() to None) 2017-11-02 11:25:48 -07:00
session.py
session_test.py
sparse_to_dense_mask_test.py
task.py
test_util.py
text_file_reader.py
timeout_guard.py
toy_regression_test.py
tt_core.py
tt_core_test.py
utils.py
visualize.py
workspace.py Add ONNX exporter for glcgan 2017-11-14 10:09:44 -08:00
workspace_test.py Compute flops in conv based on output image size 2017-12-09 21:32:08 -08:00