saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/python
History
Yi Cheng dfa0af093d Move predictor into caffe2/caffe2/predictor (#9548) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9548

Pull Request resolved: https://github.com/pytorch/translate/pull/157

One part of refactor predictor. Move all the files into predictor dir.

Reviewed By: highker

Differential Revision: D8845276

fbshipit-source-id: 1e917464b0c8a042f025128a082c784eaa3b7013
2018-07-26 19:03:40 -07:00
..
docs
examples Update from Facebook (#6692) 2018-04-17 23:36:40 -07:00
helpers [Caffe2] Add full impl of GroupNorm (#7058) 2018-04-29 11:26:40 -07:00
ideep Add fallback to TensorCPU if there are unsupported types for IDEEP Tensor (#9667) 2018-07-23 13:54:57 -07:00
layers Co-disitillation with different archs and/or feature set (#9793) 2018-07-25 10:10:27 -07:00
mint
mkl Workaround lack of 0-dim support in ideep (#8959) 2018-06-27 20:24:28 -07:00
modeling fix the annotation 2018-07-12 18:53:59 -07:00
models [Caffe2] Revamp the convnet benchmark code by using models from model zoo (#7351) 2018-05-08 08:53:52 -07:00
onnx Pass shape infos to ONNX -> Caffe2 C++ conversion backend (#9870) 2018-07-26 12:00:32 -07:00
operator_test Simplify order switch operators (#9581) 2018-07-26 18:24:29 -07:00
predictor Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
rnn
test Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
trt ONNXIFI transform (#9569) 2018-07-20 15:09:59 -07:00
__init__.py
_import_c_extension.py HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
allcompare_test.py
attention.py [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
benchmark_generator.py
binarysize.py
brew.py [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -07:00
brew_test.py Move tanh function to math (#9328) 2018-07-11 13:59:50 -07:00
build.py
cached_reader.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
caffe_translator.py
caffe_translator_test.py
checkpoint.py Update from facebook (#6871) 2018-04-23 15:01:56 -07:00
checkpoint_test.py Update from facebook (#6871) 2018-04-23 15:01:56 -07:00
CMakeLists.txt caffe2 hip python binding (#8491) 2018-06-14 19:56:56 -07:00
cnn.py [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -07:00
context.py Resolve name conflict of ContextManager (#7244) 2018-06-22 00:41:51 -04:00
context_test.py
control.py
control_ops_grad.py
control_ops_util.py
control_test.py
convnet_benchmarks.py
convnet_benchmarks_test.py
core.py set_input_record() should always add external input (#9636) 2018-07-20 11:55:37 -07:00
core_gradients_test.py [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
core_test.py set_input_record() should always add external input (#9636) 2018-07-20 11:55:37 -07:00
crf.py
data_parallel_model.py Modify barrier net to allow better control over its initialization and execution in DPM (#9665) 2018-07-22 00:23:47 -07:00
data_parallel_model_test.py Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
data_workers.py Update from facebook (#7696) 2018-05-19 23:10:48 -07:00
data_workers_test.py
dataio.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
dataio_test.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
dataset.py Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
db_file_reader.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
db_test.py
device_checker.py Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
dlpack.h
dyndep.py
embedding_generation_benchmark.py
experiment_util.py
extension_loader.py
functional.py Add support for specifying device_option in Functional (#9619) 2018-07-24 14:41:59 -07:00
functional_test.py Add support for specifying device_option in Functional (#9619) 2018-07-24 14:41:59 -07:00
fused_8bit_rowwise_conversion_ops_test.py
gradient_check_test.py [Caffe2] Fix gradient_check on in-place ops (#8828) 2018-06-25 15:25:56 -07:00
gradient_checker.py [Caffe2] Fix gradient_check on in-place ops (#8828) 2018-06-25 15:25:56 -07:00
gru_cell.py
hsm_util.py
hypothesis_test.py Temporarily disable test_unique on rocm since it keeps running into segfault (#9872) 2018-07-26 08:34:00 -07:00
hypothesis_test_util.py Temporarily disable test_unique on rocm since it keeps running into segfault (#9872) 2018-07-26 08:34:00 -07:00
ideep_test_util.py [feature request] [Caffe2] Enable MKLDNN support for inference (#6699) 2018-04-22 21:58:14 -07:00
layer_model_helper.py Improving diagnose RF NE with Cali (#9550) 2018-07-23 13:40:21 -07:00
layer_model_instantiator.py
layer_parameter_sharing_test.py
layer_test_util.py
layers_test.py fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af 2018-06-27 04:50:56 -07:00
lengths_reducer_fused_8bit_rowwise_ops_test.py
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 Adadelta optimizer to caffe2 (#9088) 2018-07-24 20:09:21 -07:00
modifier_context.py
mpi_python.cc
muji.py [Caffe2] Support non peer access in muji and fix bug when reduced_affix is empty (#6896) 2018-06-05 03:14:43 +08:00
muji_test.py [Caffe2] Support non peer access in muji and fix bug when reduced_affix is empty (#6896) 2018-06-05 03:14:43 +08:00
net_builder.py Update from Facebook (#6692) 2018-04-17 23:36:40 -07:00
net_builder_test.py
net_drawer.py
net_printer.py [DT] [38/n] Rename add_stop_signal to add_stop_condition (#6825) 2018-04-23 10:39:37 -07:00
net_printer_test.py
normalizer.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
normalizer_context.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
normalizer_test.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
numa_benchmark.py Update from facebook (#6871) 2018-04-23 15:01:56 -07:00
numa_test.py
observer_test.py
optimizer.py Add Adadelta optimizer to caffe2 (#9088) 2018-07-24 20:09:21 -07:00
optimizer_context.py
optimizer_test.py Add Adadelta optimizer to caffe2 (#9088) 2018-07-24 20:09:21 -07:00
optimizer_test_util.py Implementation of Wngrad optimizer caffe2 python wrapper and unit test on least square regression (#9001) 2018-07-13 18:54:52 -07:00
parallel_workers.py Update from facebook (#7696) 2018-05-19 23:10:48 -07:00
parallel_workers_test.py
parallelize_bmuf_distributed_test.py
pipeline.py
pipeline_test.py
predictor_constants.py
pybind_state.cc Move predictor into caffe2/caffe2/predictor (#9548) 2018-07-26 19:03:40 -07:00
pybind_state.h Revert D8121878: Remove template parameter from Tensor 2018-07-26 14:02:04 -07:00
pybind_state_dlpack.cc
pybind_state_dlpack.h Revert D8121878: Remove template parameter from Tensor 2018-07-26 14:02:04 -07:00
pybind_state_gpu.cc Revert D8121878: Remove template parameter from Tensor 2018-07-26 14:02:04 -07:00
pybind_state_hip.cc Revert D8121878: Remove template parameter from Tensor 2018-07-26 14:02:04 -07:00
pybind_state_ideep.cc Fix build error in pybind_state_ideep (#8684) 2018-06-20 08:29:48 -07:00
pybind_state_int8.cc Revert D8121878: Remove template parameter from Tensor 2018-07-26 14:02:04 -07:00
pybind_state_mkl.cc
python_op_test.py
queue_util.py
record_queue.py
recurrent.py
regularizer.py Add GroupL1Norm regularizer (#9115) 2018-07-06 13:26:09 -07:00
regularizer_context.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
regularizer_test.py Add GroupL1Norm regularizer (#9115) 2018-07-06 13:26:09 -07:00
rnn_cell.py Fix typo in comment (#7183) 2018-05-02 11:58:30 -07:00
schema.py Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
schema_test.py Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
scope.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
scope_test.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
session.py
session_test.py
sparse_to_dense_mask_test.py
sparse_to_dense_test.py
task.py
test_util.py
text_file_reader.py
timeout_guard.py
toy_regression_test.py
transformations.py Enable Conv fusion optimizations in optimizeForIdeep (#9255) 2018-07-16 21:28:50 -07:00
transformations_test.py fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af 2018-06-27 04:50:56 -07:00
tt_core.py
tt_core_test.py
utils.py Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
visualize.py
workspace.py Add interface to provide blob types to shape&type inference (#9643) 2018-07-24 11:58:05 -07:00
workspace_test.py Add workspace.RunPlanInBackground (#9637) 2018-07-20 14:56:12 -07:00