| .. |
|
benchmarks
|
|
|
|
docs
|
|
|
|
examples
|
|
|
|
fakelowp
|
|
|
|
helpers
|
|
|
|
ideep
|
|
|
|
layers
|
|
|
|
mint
|
|
|
|
mkl
|
|
|
|
modeling
|
|
|
|
models
|
|
|
|
onnx
|
[ONNX] Update onnx submodule to 1.12 (#79585)
|
2022-06-23 18:09:07 +00:00 |
|
operator_test
|
Disable GPU tests for the PiecewiseLinearTransform operator. (#75738)
|
2022-04-14 12:07:50 +00:00 |
|
predictor
|
clear predict_net field from PredictorExporterMeta stored in the exporter to save memory (#68485)
|
2021-12-10 21:25:36 -08:00 |
|
rnn
|
|
|
|
serialized_test
|
[typing] suppress errors in fbcode/caffe2 - batch 2
|
2021-08-10 10:26:52 -07:00 |
|
test
|
|
|
|
trt
|
Fix pretrained=True for test_pt_onnx_trt (#67818)
|
2021-11-04 09:49:42 -07:00 |
|
__init__.py
|
Add warning when importing caffe2 on build without BUILD_CAFFE2=1
|
2022-04-21 12:28:10 +00:00 |
|
_import_c_extension.py
|
|
|
|
_import_c_extension.pyi
|
[caffe2] fix type annotations for workspace.SwitchWorkspace() (#77464)
|
2022-05-20 19:22:06 +00:00 |
|
allcompare_test.py
|
|
|
|
attention.py
|
|
|
|
benchmark_generator.py
|
Fix ONNX ATen fallback for non-caffe2 engines
|
2022-04-14 23:18:45 +00:00 |
|
binarysize.py
|
|
|
|
brew.py
|
|
|
|
brew_test.py
|
|
|
|
build.py
|
|
|
|
cached_reader.py
|
|
|
|
caffe_translator.py
|
|
|
|
caffe_translator_test.py
|
|
|
|
checkpoint.py
|
Fix some typos.
|
2022-04-11 21:55:59 +00:00 |
|
checkpoint_test.py
|
|
|
|
CMakeLists.txt
|
fix torch.config can't respect USE_MKLDNN flag issue (#75001)
|
2022-07-17 15:00:48 +00:00 |
|
cnn.py
|
|
|
|
context.py
|
|
|
|
context.pyi
|
|
|
|
context_test.py
|
|
|
|
control.py
|
|
|
|
control_ops_grad.py
|
|
|
|
control_ops_grad_test.py
|
|
|
|
control_ops_util.py
|
|
|
|
control_test.py
|
|
|
|
convert.py
|
|
|
|
convert_test.py
|
|
|
|
convnet_benchmarks.py
|
|
|
|
convnet_benchmarks_test.py
|
|
|
|
core.py
|
[easy] add an inplace argument to MutableNetProto.to_net() and core.Net() constructor (#63068)
|
2021-08-11 11:10:52 -07:00 |
|
core_gradients_test.py
|
Fix some typos.
|
2022-04-11 21:55:59 +00:00 |
|
core_test.py
|
Fix asserts in tests (#72864)
|
2022-02-16 18:35:16 +00:00 |
|
crf.py
|
|
|
|
crf_predict.py
|
|
|
|
crf_viterbi_test.py
|
|
|
|
data_parallel_model.py
|
|
|
|
data_parallel_model_test.py
|
|
|
|
data_workers.py
|
|
|
|
data_workers_test.py
|
|
|
|
dataio.py
|
|
|
|
dataio_test.py
|
|
|
|
dataset.py
|
|
|
|
db_file_reader.py
|
|
|
|
db_test.py
|
|
|
|
device_checker.py
|
|
|
|
dlpack.h
|
Revert "Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack. (#78154)"
|
2022-07-05 08:52:46 +00:00 |
|
dyndep.py
|
|
|
|
embedding_generation_benchmark.py
|
|
|
|
experiment_util.py
|
|
|
|
extension_loader.py
|
|
|
|
fakefp16_transform_lib.py
|
|
|
|
filler_test.py
|
|
|
|
functional.py
|
|
|
|
functional_test.py
|
|
|
|
fused_8bit_rowwise_conversion_ops_test.py
|
|
|
|
gradient_check_test.py
|
|
|
|
gradient_checker.py
|
|
|
|
gru_cell.py
|
Fix some typos.
|
2022-04-11 21:55:59 +00:00 |
|
hip_test_util.py
|
|
|
|
hsm_util.py
|
|
|
|
hypothesis_test.py
|
|
|
|
hypothesis_test_util.py
|
|
|
|
ideep_test_util.py
|
|
|
|
layer_model_helper.py
|
|
|
|
layer_model_instantiator.py
|
|
|
|
layer_parameter_sharing_test.py
|
|
|
|
layer_test_util.py
|
|
|
|
layers_test.py
|
|
|
|
lazy.py
|
|
|
|
lazy_dyndep.py
|
|
|
|
lazy_dyndep_test.py
|
|
|
|
lengths_reducer_fused_8bit_rowwise_ops_test.py
|
|
|
|
lengths_reducer_rowwise_8bit_ops_test.py
|
|
|
|
lstm_benchmark.py
|
|
|
|
memonger.py
|
Remove trailing semicolon. (#74031)
|
2022-03-13 16:25:42 +00:00 |
|
memonger_test.py
|
|
|
|
mkl_test_util.py
|
|
|
|
model_device_test.py
|
|
|
|
model_helper.py
|
|
|
|
model_helper_test.py
|
|
|
|
modifier_context.py
|
|
|
|
mpi_python.cc
|
|
|
|
muji.py
|
|
|
|
muji_test.py
|
|
|
|
net_builder.py
|
|
|
|
net_builder_test.py
|
|
|
|
net_drawer.py
|
|
|
|
net_printer.py
|
|
|
|
net_printer_test.py
|
|
|
|
nomnigraph.py
|
|
|
|
nomnigraph_test.py
|
|
|
|
nomnigraph_transformations.py
|
|
|
|
nomnigraph_transformations_test.py
|
|
|
|
normalizer.py
|
|
|
|
normalizer_context.py
|
|
|
|
normalizer_test.py
|
|
|
|
numa_benchmark.py
|
|
|
|
numa_test.py
|
|
|
|
observer_test.py
|
|
|
|
operator_fp_exceptions_test.py
|
|
|
|
optimizer.py
|
reduce plan generation log spam (#70880)
|
2022-01-08 10:07:06 -08:00 |
|
optimizer_context.py
|
|
|
|
optimizer_test.py
|
|
|
|
optimizer_test_util.py
|
|
|
|
parallel_workers.py
|
|
|
|
parallel_workers_test.py
|
|
|
|
parallelize_bmuf_distributed_test.py
|
|
|
|
pipeline.py
|
|
|
|
pipeline_test.py
|
|
|
|
predictor_constants.py
|
|
|
|
pybind_state.cc
|
fix torch.config can't respect USE_MKLDNN flag issue (#75001)
|
2022-07-17 15:00:48 +00:00 |
|
pybind_state.h
|
use irange for loops (#70248)
|
2022-01-06 23:14:29 -08:00 |
|
pybind_state_dlpack.cc
|
Bump dlpack.h to latest version (#65047)
|
2022-01-21 16:55:14 +00:00 |
|
pybind_state_dlpack.h
|
Bump dlpack.h to latest version (#65047)
|
2022-01-21 16:55:14 +00:00 |
|
pybind_state_gpu.cc
|
[NOOP][clangformat][codemod] Enable CLANGFORMAT for caffe2/caffe2/* (#67624)
|
2021-11-02 22:14:04 -07:00 |
|
pybind_state_hip.cc
|
[NOOP][clangformat][codemod] Enable CLANGFORMAT for caffe2/caffe2/* (#67624)
|
2021-11-02 22:14:04 -07:00 |
|
pybind_state_ideep.cc
|
Update fbcode symlinks for mkl-dnn ideep 2.5.2
|
2022-03-04 06:40:08 +00:00 |
|
pybind_state_int8.cc
|
|
|
|
pybind_state_nomni.cc
|
[NOOP][clangformat][codemod] Enable CLANGFORMAT for caffe2/caffe2/* (#67624)
|
2021-11-02 22:14:04 -07:00 |
|
pybind_state_registry.cc
|
|
|
|
pybind_state_registry.h
|
|
|
|
python_op_test.py
|
|
|
|
queue_util.py
|
|
|
|
record_queue.py
|
|
|
|
recurrent.py
|
docs: Fix a few typos (#81435)
|
2022-07-14 04:20:26 +00:00 |
|
regularizer.py
|
|
|
|
regularizer_context.py
|
|
|
|
regularizer_test.py
|
|
|
|
rnn_cell.py
|
|
|
|
schema.py
|
|
|
|
schema_test.py
|
|
|
|
scope.py
|
|
|
|
scope_test.py
|
|
|
|
session.py
|
|
|
|
session_test.py
|
|
|
|
sparse_to_dense_mask_test.py
|
|
|
|
sparse_to_dense_test.py
|
|
|
|
task.py
|
|
|
|
task_test.py
|
|
|
|
test_util.py
|
|
|
|
text_file_reader.py
|
|
|
|
timeout_guard.py
|
|
|
|
toy_regression_test.py
|
|
|
|
transformations.py
|
|
|
|
transformations_test.py
|
|
|
|
tt_core.py
|
|
|
|
tt_core_test.py
|
|
|
|
utils.py
|
|
|
|
utils_test.py
|
|
|
|
visualize.py
|
|
|
|
workspace.py
|
|
|
|
workspace_test.py
|
[codemod][type-comments] Convert type comments in workspace_test.py (#73086)
|
2022-02-18 22:36:25 +00:00 |
