frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
bb9ff58c6d52ba47688e5bfa2021cac8fa94bd64
pytorch/caffe2/python
History
Xiaomeng Yang bb9ff58c6d Add cudnn activation ops (#9379) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9379

Add cudnn activation ops

Reviewed By: houseroad

Differential Revision: D8818013

fbshipit-source-id: d3881c634a46578b9331da07f9fdf7e1f31d7e8a
2018-07-12 23:18:56 -07:00
..
docs
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
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
Normalize IDEEP spatial bn op test (#9276)
2018-07-09 11:55:41 -07:00
layers
fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af
2018-06-27 04:50:56 -07:00
mint
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
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
Skip the count_include_pad in average pool for now (#9365)
2018-07-11 18:09:50 -07:00
operator_test
Add cudnn activation ops (#9379)
2018-07-12 23:18:56 -07:00
predictor
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
rnn
Fix a typo
2018-03-30 21:00:44 -07:00
test
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
trt
Fix TensorRT tests (#9285)
2018-07-09 20:40:19 -07:00
__init__.py
Experimental support for setup.py develop mode install
2018-02-12 23:36:18 -08:00
_import_c_extension.py
caffe2 hip python binding (#8491)
2018-06-14 19:56:56 -07:00
allcompare_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
attention.py
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -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
Move tanh function to math (#9328)
2018-07-11 13:59:50 -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
Update from Facebook (#8887)
2018-06-26 14:55:48 -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 from facebook (#6871)
2018-04-23 15:01:56 -07:00
checkpoint.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_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
context.py
Resolve name conflict of ContextManager (#7244)
2018-06-22 00:41:51 -04: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
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
core_test.py
Update from facebook 1ee4edd286a3 (#8040)
2018-06-01 17:41:09 -04:00
core.py
Make error message more informative (#9352)
2018-07-11 15:09:46 -07:00
crf.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
data_parallel_model_test.py
Update from facebook (#7451)
2018-05-10 23:14:27 -07:00
data_parallel_model.py
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
data_workers_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
data_workers.py
Update from facebook (#7696)
2018-05-19 23:10:48 -07:00
dataio_test.py
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
dataio.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
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
device_checker.py
Update from facebook (#7451)
2018-05-10 23:14:27 -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
[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
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
[Caffe2] Fix gradient_check on in-place ops (#8828)
2018-06-25 15:25:56 -07:00
hypothesis_test.py
Add FTRL Optimzier with Group Lasso regularizer (#9074)
2018-07-06 13:41: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
fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af
2018-06-27 04:50:56 -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
fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af
2018-06-27 04:50:56 -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
[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.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_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
net_builder.py
Update from Facebook (#6692)
2018-04-17 23:36:40 -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
[DT] [38/n] Rename add_stop_signal to add_stop_condition (#6825)
2018-04-23 10:39:37 -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
normalizer.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
[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
Add FTRL Optimzier with Group Lasso regularizer (#9074)
2018-07-06 13:41:00 -07:00
optimizer.py
Add FTRL Optimzier with Group Lasso regularizer (#9074)
2018-07-06 13:41:00 -07:00
parallel_workers_test.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
parallel_workers.py
Update from facebook (#7696)
2018-05-19 23:10:48 -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
TypeId improvements (#8350)
2018-06-14 09:16:26 -07:00
pybind_state_gpu.cc
[Caffe2] Provide option to initialize the TensorRT engine at Operator constructor time (#6809)
2018-04-23 13:09:35 -07:00
pybind_state_hip.cc
[Caffe2] Enable AMD/MIOPEN ops for Caffe2 (#8306)
2018-06-13 04:00:39 -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
fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af
2018-06-27 04:50:56 -07:00
pybind_state_mkl.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
pybind_state.cc
fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af
2018-06-27 04:50:56 -07:00
pybind_state.h
Fix the build break for python3.7 PyUnicode_AsUTF8AndSize() prototype changing (#9259)
2018-07-11 10:39:43 -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
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
regularizer.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_test.py
Update from facebook (#7855)
2018-05-29 11:38:02 -07:00
schema.py
Update from facebook (#7855)
2018-05-29 11:38:02 -07:00
scope_test.py
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
scope.py
Update from Facebook (#8887)
2018-06-26 14:55:48 -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
fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af
2018-06-27 04:50:56 -07:00
transformations.py
fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af
2018-06-27 04:50:56 -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
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
visualize.py
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
workspace_test.py
[feature request] [Caffe2] Enable MKLDNN support for inference (#6699)
2018-04-22 21:58:14 -07:00
workspace.py
Update from facebook (#7696)
2018-05-19 23:10:48 -07:00