pytorch/caffe2/python
Deepak Gopinath 6aee34b666 Registering GPU version of PackSegments using GPUFallbackOp
Summary: Creating PackSegments and UnpackSegments GPU operators using GPUFallbackOp for now. The op does mainly copying of blobs and this is a reasonable solution until we have a CUDA op.

Reviewed By: pietern

Differential Revision: D4761589

fbshipit-source-id: dd483b9e34ecb6b53925405e5b4c24859c549606
2017-03-24 16:01:53 -07:00
..
docs
examples Remove more instances of batch_size 2017-03-19 22:31:30 -07:00
layers allows to specify output names for functional layers 2017-03-23 13:18:58 -07:00
mint
models exec_net --> predict_net 2017-03-23 16:31:49 -07:00
operator_test Registering GPU version of PackSegments using GPUFallbackOp 2017-03-24 16:01:53 -07:00
_import_c_extension.py
attention.py Removing unnecessary Copy after fixing gradients for external parameters 2017-03-23 15:04:22 -07:00
caffe_translator.py translator fix 2017-03-22 15:09:13 -07:00
caffe_translator_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
checkpoint.py
checkpoint_test.py
CMakeLists.txt
cnn.py Add ConvNd to model helper 2017-03-22 15:47:48 -07:00
context.py
context_test.py
control.py
control_test.py
convnet_benchmarks.py
convnet_benchmarks_test.py
core.py fixes to make data parallel model work for RecurrentNet + test case 2017-03-14 15:48:07 -07:00
core_gradients_test.py
core_test.py
crf.py CRF layer in caffe2 2017-03-23 22:02:02 -07:00
data_parallel_model.py add gpu support for caffe2-seq2seq 2017-03-17 05:19:14 -07:00
data_parallel_model_test.py fixes to make data parallel model work for RecurrentNet + test case 2017-03-14 15:48:07 -07:00
data_workers.py
data_workers_test.py
dataio.py Stop multi_reader if we run out of data before max_examples 2017-03-10 18:03:57 -08:00
dataio_test.py Stop multi_reader if we run out of data before max_examples 2017-03-10 18:03:57 -08:00
dataset.py
db_test.py
device_checker.py
dyndep.py
experiment_util.py fix external_loggers 2017-03-16 13:47:58 -07:00
extension_loader.py
gradient_check_test.py
gradient_checker.py
hsm_util.py
hypothesis_test.py
hypothesis_test_util.py
introspect_vis.py
layer_model_helper.py Add Calibration 2017-03-24 14:32:23 -07:00
layer_model_instantiator.py
layers_test.py allows to specify output names for functional layers 2017-03-23 13:18:58 -07:00
load_save_test.py Improve error message from LogFileDB on missing file 2017-03-10 23:31:28 -08:00
lstm_benchmark.py
memonger.py
memonger_test.py
mkl_test_util.py
model_device_test.py
model_helper.py add gpu support for caffe2-seq2seq 2017-03-17 05:19:14 -07:00
mpi_python.cc
muji.py
muji_test.py
net_builder.py
net_builder_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
net_drawer.py
net_printer.py
net_printer_test.py
optimizer.py
optimizer_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
optimizer_test_util.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
pipeline.py
pybind_state.cc Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00
pybind_state.h bugfix for Windows, esp. VS 2017 2017-03-21 05:17:59 -07:00
pybind_state_gpu.cc
pybind_state_mkl.cc
python_op_test.py
queue_util.py More perf stats for BlobsQueue 2017-03-24 14:03:28 -07:00
record_queue.py
recurrent.py Don't do copy for param_grad in backward_step_net 2017-03-24 02:22:33 -07:00
schema.py migrate mtml to dper2 2017-03-16 17:48:05 -07:00
schema_test.py support get/add a field by nested name 2017-03-15 02:00:57 -07:00
scope.py
scope_test.py
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 Extract net and blobs assignment to separate functions 2017-03-24 13:17:24 -07:00
visualize.py
workspace.py Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00
workspace_test.py Added predictor bindings to python interface 2017-03-15 11:17:54 -07:00