Ahmed Taei a0fe96d7cd Rewrite memonger DAG in C++. ... Summary: This diff replaces the main of the memonger for dag algorithm _compute_blob_recycling_for_dag with a c++ implementation. Reviewed By: akyrola Differential Revision: D5544219 fbshipit-source-id: 9f868880c8d0eb997ad3dd39433f9d0b9216d303		2017-08-16 16:17:15 -07:00
..
docs
examples	Important typo in resnet50_trainer	2017-08-15 19:03:15 -07:00
helpers	Add TensorCore support	2017-08-10 20:16:48 -07:00
layers	Make reservoir sampling thread safe	2017-08-10 15:27:21 -07:00
mint
mkl	Support grouped convolutions in MKL	2017-07-25 14:19:02 -07:00
modeling
models	rectify args btw. train and translate	2017-08-10 15:27:18 -07:00
operator_test	Caffe2: diagonal fill op	2017-08-16 13:05:11 -07:00
predictor
rnn	Revert D5589309: modify _LSTM into _RNN to adapt GRU	2017-08-10 16:42:41 -07:00
_import_c_extension.py
allcompare_test.py	Adding AllCompare-like function to data_parallel_model	2017-07-13 13:03:57 -07:00
attention.py	Reduce memory usage for dot attention	2017-08-14 12:35:50 -07:00
binarysize.py	binary size util	2017-07-14 17:49:24 -07:00
brew.py
brew_test.py	Allow passing unsymmetric 2d kernels to brew.conv.	2017-08-10 15:27:16 -07:00
caffe_translator.py
caffe_translator_test.py
checkpoint.py	Temporarily disables the checkpoints for the readers.	2017-08-15 19:36:11 -07:00
checkpoint_test.py	Fixes the broken checkpoint test.	2017-08-16 11:24:50 -07:00
CMakeLists.txt
cnn.py
context.py
context_test.py
control.py
control_test.py
convnet_benchmarks.py
convnet_benchmarks_test.py
core.py	Support session in distributed realtime trainer	2017-08-16 10:28:55 -07:00
core_gradients_test.py	warn about orphan StopGradient output	2017-07-20 21:41:41 -07:00
core_test.py	Support session in distributed realtime trainer	2017-08-16 10:28:55 -07:00
crf.py
data_parallel_model.py	DataParallelModel: take param_init_net into account in _InferBlobDevice	2017-08-15 12:06:46 -07:00
data_parallel_model_test.py	disable travis test for dpm test	2017-08-15 19:17:41 -07:00
data_workers.py	Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py	2017-08-07 10:14:08 -07:00
data_workers_test.py	Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py	2017-08-07 10:14:08 -07:00
dataio.py
dataio_test.py
dataset.py	Option to enforce batch size	2017-08-01 22:29:55 -07:00
db_test.py
device_checker.py
dyndep.py
embedding_generation_benchmark.py	Benchmark for embedding generation	2017-08-15 14:22:41 -07:00
empty.so
experiment_util.py
extension_loader.py
gradient_check_test.py
gradient_checker.py
gru_cell.py	Revert D5589309: modify _LSTM into _RNN to adapt GRU	2017-08-10 16:42:41 -07:00
hsm_util.py
hypothesis_test.py	Implement gradients for Col2Im and Im2Col operators	2017-08-07 15:51:30 -07:00
hypothesis_test_util.py	Add CUDA implementation of BooleanUnmask and fixed some bugs in the test	2017-08-01 16:51:40 -07:00
layer_model_helper.py	Adding parameter sharing API to Dper2	2017-08-03 00:33:18 -07:00
layer_model_instantiator.py	saving/loading CPU/GPU nets	2017-07-23 02:18:15 -07:00
layer_parameter_sharing_test.py	Adding parameter sharing API to Dper2	2017-08-03 00:33:18 -07:00
layer_test_util.py	Add a method to run a train net multiple times in layer_test_util.py	2017-07-28 19:56:05 -07:00
layers_test.py	Add conv layer and layer tests	2017-08-08 10:57:43 -07:00
load_save_test.py
lstm_benchmark.py	Revert D5001637: [Caffe2][RNN] Threaded dependency-aware RNNExecutor (frontier/diagonal execution).	2017-08-16 03:21:49 -07:00
memonger.py	Rewrite memonger DAG in C++.	2017-08-16 16:17:15 -07:00
memonger_test.py	fix for duplicate input case	2017-07-13 01:51:30 -07:00
mkl_test_util.py	Implement a filler op test	2017-07-25 14:18:57 -07:00
model_device_test.py
model_helper.py	ExtractPredictorNet should strip gpu_id prefix from step_net	2017-07-27 16:06:47 -07:00
mpi_python.cc
muji.py
muji_test.py
net_builder.py
net_builder_test.py
net_drawer.py
net_printer.py	net_printer.to_string() accepts NetDef	2017-08-01 10:17:29 -07:00
net_printer_test.py
optimizer.py	Populate learning rate blob name into data_parallel_model and fix resnet50_trainer example.	2017-07-21 16:24:10 -07:00
optimizer_context.py
optimizer_test.py	Always use assertAlmostEqual for floats when crossing python and C boundaries	2017-08-06 14:51:11 -07:00
optimizer_test_util.py
parallel_workers.py	Caffe2 [easy]: Better exception logging in parallel_workers/data_workers	2017-08-10 15:27:19 -07:00
parallel_workers_test.py	Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py	2017-08-07 10:14:08 -07:00
parallelize_gpu_bmuf_distributed_test.py	Added Nesterov	2017-08-11 13:52:43 -07:00
pipeline.py
predictor_constants.py
pybind_state.cc	Rewrite memonger DAG in C++.	2017-08-16 16:17:15 -07:00
pybind_state.h
pybind_state_gpu.cc
pybind_state_mkl.cc	MKL code move	2017-07-26 20:21:55 -07:00
python_op_test.py
queue_util.py	remove unsed code and bring back single benchmark mode	2017-08-16 14:06:31 -07:00
record_queue.py
recurrent.py	Revert D5001637: [Caffe2][RNN] Threaded dependency-aware RNNExecutor (frontier/diagonal execution).	2017-08-16 03:21:49 -07:00
rnn_cell.py	Revert D5589309: modify _LSTM into _RNN to adapt GRU	2017-08-10 16:42:41 -07:00
schema.py	Return empty Struct when get_field has empty input	2017-08-01 19:49:47 -07:00
schema_test.py	Return empty Struct when get_field has empty input	2017-08-01 19:49:47 -07:00
scope.py
scope_test.py
session.py	Support session in distributed realtime trainer	2017-08-16 10:28:55 -07:00
session_test.py
sparse_to_dense_mask_test.py
task.py	Support session in distributed realtime trainer	2017-08-16 10:28:55 -07:00
test_util.py
text_file_reader.py
timeout_guard.py
toy_regression_test.py
tt_core.py
tt_core_test.py
utils.py	include numpy's other 32bit int type	2017-08-01 13:53:11 -07:00
visualize.py
workspace.py	Transforms in Python	2017-08-01 16:51:38 -07:00
workspace_test.py	Transforms in Python	2017-08-01 16:51:38 -07:00