| .. |
|
benchmarks
|
|
|
|
docs
|
Change caffe2 branch links to main (#100129)
|
2023-04-27 10:31:50 +00:00 |
|
examples
|
|
|
|
fakelowp
|
|
|
|
helpers
|
|
|
|
ideep
|
[CODEMOD][caffe2] replace numpy.bool with bool (#111432)
|
2023-10-18 18:56:40 +00:00 |
|
layers
|
|
|
|
mint
|
|
|
|
mkl
|
|
|
|
modeling
|
[CODEMOD][caffe2] replace uses of np.float with np.float64 (#112675)
|
2023-11-03 03:00:51 +00:00 |
|
models
|
[BE]: enable ruff rules PLR1722 and PLW3301 (#109461)
|
2023-09-18 02:07:21 +00:00 |
|
onnx
|
|
|
|
operator_test
|
[CODEMOD][caffe2] replace uses of np.float with np.float64 (#112675)
|
2023-11-03 03:00:51 +00:00 |
|
predictor
|
[BE]: Cleanup deprecated stdlib imports (UP006,UP035) (#101361)
|
2023-05-15 14:32:41 +00:00 |
|
rnn
|
|
|
|
serialized_test
|
[caffe2] replace numpy.object with object (#111494)
|
2023-10-19 04:37:00 +00:00 |
|
test
|
Fix possible naming collision issue (#107743)
|
2023-09-08 17:39:27 +00:00 |
|
trt
|
|
|
|
__init__.py
|
Revert "[Reland2] Update NVTX to NVTX3 (#109843)"
|
2023-12-05 16:10:20 +00:00 |
|
_import_c_extension.py
|
|
|
|
_import_c_extension.pyi
|
[BE]: Apply PYI autofixes to various types (#107521)
|
2023-08-20 02:42:21 +00:00 |
|
allcompare_test.py
|
|
|
|
attention.py
|
|
|
|
benchmark_generator.py
|
|
|
|
binarysize.py
|
|
|
|
brew.py
|
|
|
|
brew_test.py
|
|
|
|
build.py
|
|
|
|
cached_reader.py
|
|
|
|
caffe_translator.py
|
|
|
|
caffe_translator_test.py
|
|
|
|
checkpoint.py
|
|
|
|
checkpoint_test.py
|
|
|
|
clean_workspace_test.py
|
|
|
|
CMakeLists.txt
|
|
|
|
cnn.py
|
|
|
|
context.py
|
|
|
|
context.pyi
|
[BE] Enable Ruff's Flake8 PYI001 (#112823)
|
2023-11-03 17:25:39 +00:00 |
|
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
|
[CODEMOD][caffe2] replace numpy.bool with bool (#111432)
|
2023-10-18 18:56:40 +00:00 |
|
core_gradients_test.py
|
|
|
|
core_test.py
|
[CODEMOD][caffe2] replace uses of np.float with np.float64 (#112675)
|
2023-11-03 03:00:51 +00:00 |
|
crf.py
|
|
|
|
crf_predict.py
|
|
|
|
crf_viterbi_test.py
|
|
|
|
data_parallel_model.py
|
|
|
|
data_parallel_model_test.py
|
[Codemod][[codemod] Replace third-party mock with unittest.mock] caffe2/caffe2 (#106541)
|
2023-09-29 18:09:49 +00:00 |
|
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
|
Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack … (#94968)
|
2023-03-30 04:32:15 +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
|
Fix fused_8bit_rowwise_conversion_ops_test (#98183)
|
2023-04-04 16:06:13 +00:00 |
|
gradient_check_test.py
|
|
|
|
gradient_checker.py
|
|
|
|
gru_cell.py
|
|
|
|
hip_test_util.py
|
|
|
|
hsm_util.py
|
|
|
|
hypothesis_test.py
|
[CODEMOD][caffe2] replace numpy.bool with bool (#111432)
|
2023-10-18 18:56:40 +00:00 |
|
hypothesis_test_util.py
|
[CODEMOD][caffe2] replace numpy.bool with bool (#111432)
|
2023-10-18 18:56:40 +00:00 |
|
ideep_test_util.py
|
|
|
|
layer_model_helper.py
|
[CODEMOD][caffe2] replace uses of np.float with np.float64 (#112675)
|
2023-11-03 03:00:51 +00:00 |
|
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
|
[BE]: Remove useless lambdas (#113602)
|
2023-11-14 20:06:48 +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
|
Enable flake8-bugbear B020 lint (#110823)
|
2023-10-24 22:43:47 +00:00 |
|
net_drawer.py
|
[BE]: Remove useless lambdas (#113602)
|
2023-11-14 20:06:48 +00:00 |
|
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
|
[fix] Disable EMA if ema_alpha is set to None (#98992)
|
2023-04-18 17:28:07 +00:00 |
|
optimizer_context.py
|
|
|
|
optimizer_test.py
|
|
|
|
optimizer_test_util.py
|
|
|
|
parallel_workers.py
|
|
|
|
parallel_workers_test.py
|
[BE]: Remove useless lambdas (#113602)
|
2023-11-14 20:06:48 +00:00 |
|
parallelize_bmuf_distributed_test.py
|
|
|
|
pipeline.py
|
|
|
|
pipeline_test.py
|
|
|
|
predictor_constants.py
|
|
|
|
pybind_state.cc
|
Fix typo under caffe2 directory (#110825)
|
2023-10-08 20:48:12 +00:00 |
|
pybind_state.h
|
[Cmake] Check that gcc-9.4 or newer is used (#112858)
|
2023-11-06 17:19:53 +00:00 |
|
pybind_state_dlpack.cc
|
|
|
|
pybind_state_dlpack.h
|
|
|
|
pybind_state_gpu.cc
|
|
|
|
pybind_state_hip.cc
|
|
|
|
pybind_state_ideep.cc
|
|
|
|
pybind_state_int8.cc
|
|
|
|
pybind_state_nomni.cc
|
|
|
|
pybind_state_registry.cc
|
|
|
|
pybind_state_registry.h
|
|
|
|
pybind_workspace.cc
|
|
|
|
pybind_workspace.h
|
|
|
|
python_op_test.py
|
|
|
|
queue_util.py
|
|
|
|
record_queue.py
|
|
|
|
recurrent.py
|
|
|
|
regularizer.py
|
|
|
|
regularizer_context.py
|
|
|
|
regularizer_test.py
|
|
|
|
rnn_cell.py
|
|
|
|
schema.py
|
Enable flake8-bugbear B020 lint (#110823)
|
2023-10-24 22:43:47 +00:00 |
|
schema_test.py
|
[codemod][numpy] replace np.str with str (#103931)
|
2023-06-21 18:16:42 +00:00 |
|
scope.py
|
|
|
|
scope_test.py
|
|
|
|
session.py
|
|
|
|
session_test.py
|
|
|
|
sparse_to_dense_mask_test.py
|
[CODEMOD][caffe2] replace uses of np.float with np.float64 (#112675)
|
2023-11-03 03:00:51 +00:00 |
|
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
|
[codemod][numpy] replace np.str with str (#103931)
|
2023-06-21 18:16:42 +00:00 |
|
utils_test.py
|
|
|
|
visualize.py
|
|
|
|
workspace.py
|
[caffe2] replace numpy.object with object (#111494)
|
2023-10-19 04:37:00 +00:00 |
|
workspace_test.py
|
[caffe2] replace numpy.object with object (#111494)
|
2023-10-19 04:37:00 +00:00 |
