OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 52 Packages Projects Releases Wiki Activity
Files
434f710f3fc688a2df3f9723f7f2403e3c2f2494
pytorch/caffe2/python
History
Yinghai Lu 434f710f3f [Caffe2] Add support to TensorRT (#6150) 
* Add support to TensorRT

* Removed License header

* Bind input/output by position

* Comments

* More comments

* Add benchmark

* Add warning for performance degradation on large batch

* Address comments

* comments
2018-04-11 17:03:54 -07:00
..
docs
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
examples
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
helpers
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
layers
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
mint
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
mkl
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
modeling
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
models
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
onnx
[Caffe2] Add support to TensorRT (#6150)
2018-04-11 17:03:54 -07:00
operator_test
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
predictor
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
rnn
Fix a typo
2018-03-30 21:00:44 -07:00
test
Fix occasional test timeouts
2018-01-19 20:08:58 -08:00
trt
[Caffe2] Add support to TensorRT (#6150)
2018-04-11 17:03:54 -07:00
__init__.py
Experimental support for setup.py develop mode install
2018-02-12 23:36:18 -08:00
_import_c_extension.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
allcompare_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
attention.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
benchmark_generator.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
binarysize.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
brew_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
brew.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
build.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
cached_reader.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
caffe_translator_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
caffe_translator.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
checkpoint_test.py
Update the compile function of Job (#6323)
2018-04-09 22:44:23 -07:00
checkpoint.py
Update the compile function of Job (#6323)
2018-04-09 22:44:23 -07:00
CMakeLists.txt
Fix OSS build
2017-12-21 19:04:25 -08:00
cnn.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
context_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
context.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
control_ops_grad.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
control_ops_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
control_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
control.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
convnet_benchmarks_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
convnet_benchmarks.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
core_gradients_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
core_test.py
Enable support for placeholder ops in InjectCrossDeviceCopies
2018-03-27 18:10:39 -07:00
core.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
crf.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
data_parallel_model_test.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
data_parallel_model_utils.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
data_parallel_model.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
data_workers_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
data_workers.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
dataio_test.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
dataio.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
dataset.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
db_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
device_checker.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
dlpack.h
Support for DLPack in Python op
2017-12-21 17:02:16 -08:00
dyndep.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
embedding_generation_benchmark.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
experiment_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
extension_loader.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
functional_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
functional.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
fused_8bit_rowwise_conversion_ops_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
gradient_check_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
gradient_checker.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
gru_cell.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
hsm_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
hypothesis_test_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
hypothesis_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
layer_model_helper.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
layer_model_instantiator.py
[caffe2] Fbcode to GitHub sync (#6208)
2018-04-02 16:35:27 -07:00
layer_parameter_sharing_test.py
[layer model] remove duplicated init ops
2018-03-27 18:10:39 -07:00
layer_test_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
layers_test.py
[Caffe2] Changes done inside Facebook (#6378)
2018-04-10 21:11:43 -07:00
lengths_reducer_fused_8bit_rowwise_ops_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
lengths_reducer_rowwise_8bit_ops_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
lstm_benchmark.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
memonger_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
memonger.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
mkl_test_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
model_device_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
model_helper.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
modifier_context.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
mpi_python.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
muji_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
muji.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
net_builder_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
net_builder.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
net_drawer.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
net_printer_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
net_printer.py
[caffe2] use dictionary in Printer (#6443)
2018-04-10 10:37:07 -07:00
numa_benchmark.py
[core] Graph-level NUMA awareness in Caffe2
2018-03-06 00:33:11 -08:00
numa_test.py
[core] NUMA-aware pinned allocator
2018-03-06 00:33:11 -08:00
observer_test.py
Fix RNN scoping situation
2018-02-07 17:35:29 -08:00
optimizer_context.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
optimizer_test_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
optimizer_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
optimizer.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
parallel_workers_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
parallel_workers.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
parallelize_bmuf_distributed_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
pipeline_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
pipeline.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
predictor_constants.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
pybind_state_dlpack.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
pybind_state_dlpack.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
pybind_state_gpu.cc
[Caffe2] Add support to TensorRT (#6150)
2018-04-11 17:03:54 -07:00
pybind_state_int8.cc
[C2 Int8][C2 Core]fetch int8 blob
2018-03-30 21:00:44 -07:00
pybind_state_mkl.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
pybind_state.cc
Add fuseNNPACKConvRelu (#6439)
2018-04-10 16:51:16 -07:00
pybind_state.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
python_op_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
queue_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
record_queue.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
recurrent.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
regularizer_context.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
regularizer_test.py
add dense regularization
2018-01-08 13:03:17 -08:00
regularizer.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
rnn_cell.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
schema_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
schema.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
scope_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
scope.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
session_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
session.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
sparse_to_dense_mask_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
sparse_to_dense_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
task.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
test_util.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
text_file_reader.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
timeout_guard.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
toy_regression_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
transformations_test.py
Add fuseNNPACKConvRelu (#6439)
2018-04-10 16:51:16 -07:00
transformations.py
Add fuseNNPACKConvRelu (#6439)
2018-04-10 16:51:16 -07:00
tt_core_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
tt_core.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
utils.py
[layer model] remove duplicated init ops
2018-03-27 18:10:39 -07:00
visualize.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
workspace_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
workspace.py
change print to logger.warning in operator traceback code (#6216)
2018-04-03 08:01:25 -07:00