| .. |
|
docs
|
Standardize operator argument "is_test"
|
2017-09-13 14:35:27 -07:00 |
|
examples
|
for CPU-data parallel, allow sharing model
|
2017-09-15 16:19:37 -07:00 |
|
helpers
|
resubmission of PR1175: fp16 BatchMatMul
|
2017-09-14 21:46:05 -07:00 |
|
layers
|
Remove dot_product layer
|
2017-09-07 18:48:30 -07:00 |
|
mint
|
|
|
|
mkl
|
|
|
|
modeling
|
Scaled training and fetching from the PS
|
2017-08-23 18:16:03 -07:00 |
|
models
|
HTTPMessage in Python 3 does not have getheader
|
2017-09-14 13:59:06 -07:00 |
|
operator_test
|
fix cuda GatherOp for empty batch
|
2017-09-15 17:40:43 -07:00 |
|
predictor
|
Allow specification of num_workers in PredictorExportMeta and enable for NMT beam search model
|
2017-09-07 22:48:45 -07:00 |
|
rnn
|
Revert D5589309: modify _LSTM into _RNN to adapt GRU
|
2017-08-10 16:42:41 -07:00 |
|
_import_c_extension.py
|
|
|
|
allcompare_test.py
|
|
|
|
attention.py
|
resubmission of PR1175: fp16 BatchMatMul
|
2017-09-14 21:46:05 -07:00 |
|
benchmark_generator.py
|
A benchmark generator for individual ops
|
2017-08-31 17:33:21 -07:00 |
|
binarysize.py
|
|
|
|
brew.py
|
resubmission of PR1175: fp16 BatchMatMul
|
2017-09-14 21:46:05 -07:00 |
|
brew_test.py
|
Allow passing unsymmetric 2d kernels to brew.conv.
|
2017-08-10 15:27:16 -07:00 |
|
build.py
|
Strip Operator Schema in mobile build
|
2017-08-22 13:31:08 -07:00 |
|
caffe_translator.py
|
Add option to remove legacy_pad in caffe_translator
|
2017-09-14 10:32:48 -07:00 |
|
caffe_translator_test.py
|
Add option to remove legacy_pad in caffe_translator
|
2017-09-14 10:32:48 -07:00 |
|
checkpoint.py
|
Enable reader checkpoint
|
2017-09-05 14:21:25 -07:00 |
|
checkpoint_test.py
|
Changes the checkpoint naming rules.
|
2017-08-17 22:16:42 -07:00 |
|
CMakeLists.txt
|
|
|
|
cnn.py
|
|
|
|
context.py
|
Add default implementation of __call__ for context manager
|
2017-08-22 17:46:22 -07:00 |
|
context_test.py
|
Add default implementation of __call__ for context manager
|
2017-08-22 17:46:22 -07:00 |
|
control.py
|
|
|
|
control_ops_util.py
|
Control flow operators
|
2017-08-28 20:04:43 -07:00 |
|
control_test.py
|
|
|
|
convnet_benchmarks.py
|
|
|
|
convnet_benchmarks_test.py
|
|
|
|
core.py
|
fix comment on core.Net.RunAllOnMKL
|
2017-09-13 19:32:18 -07:00 |
|
core_gradients_test.py
|
tests: use assertRaises, not expectedFail
|
2017-09-13 11:39:35 -07:00 |
|
core_test.py
|
Support session in distributed realtime trainer
|
2017-08-16 10:28:55 -07:00 |
|
crf.py
|
|
|
|
data_parallel_model.py
|
for CPU-data parallel, allow sharing model
|
2017-09-15 16:19:37 -07:00 |
|
data_parallel_model_test.py
|
for CPU-data parallel, allow sharing model
|
2017-09-15 16:19:37 -07:00 |
|
data_workers.py
|
Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py
|
2017-08-07 10:14:08 -07:00 |
|
data_workers_test.py
|
Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py
|
2017-08-07 10:14:08 -07:00 |
|
dataio.py
|
Make piper of PipedReaderBuilder takes arguments
|
2017-09-08 13:46:29 -07:00 |
|
dataio_test.py
|
|
|
|
dataset.py
|
TrimDataset to ensure size is multiple of number or replicas
|
2017-09-13 12:17:21 -07:00 |
|
db_test.py
|
|
|
|
device_checker.py
|
|
|
|
dyndep.py
|
|
|
|
embedding_generation_benchmark.py
|
Benchmark for embedding generation
|
2017-08-15 14:22:41 -07:00 |
|
experiment_util.py
|
|
|
|
extension_loader.py
|
|
|
|
functional.py
|
Functional C2
|
2017-09-13 15:37:28 -07:00 |
|
functional_test.py
|
Functional C2
|
2017-09-13 15:37:28 -07:00 |
|
gradient_check_test.py
|
|
|
|
gradient_checker.py
|
|
|
|
gru_cell.py
|
Revert D5589309: modify _LSTM into _RNN to adapt GRU
|
2017-08-10 16:42:41 -07:00 |
|
hsm_util.py
|
|
|
|
hypothesis_test.py
|
EnsureDense/SparseToDense for CUDA
|
2017-09-01 09:33:05 -07:00 |
|
hypothesis_test_util.py
|
adding unit tests to compphoto caffe2 projects
|
2017-09-15 12:49:37 -07:00 |
|
layer_model_helper.py
|
Adding parameter sharing API to Dper2
|
2017-08-03 00:33:18 -07:00 |
|
layer_model_instantiator.py
|
|
|
|
layer_parameter_sharing_test.py
|
Adding parameter sharing API to Dper2
|
2017-08-03 00:33:18 -07:00 |
|
layer_test_util.py
|
Add a method to run a train net multiple times in layer_test_util.py
|
2017-07-28 19:56:05 -07:00 |
|
layers_test.py
|
Create MergeIdListsLayer
|
2017-08-22 17:00:55 -07:00 |
|
lengths_reducer_rowwise_8bit_ops_test.py
|
Rowwise quantization
|
2017-09-06 10:19:38 -07:00 |
|
load_save_test.py
|
|
|
|
lstm_benchmark.py
|
enable setting rnn executor threads and max streams
|
2017-09-08 16:36:51 -07:00 |
|
memonger.py
|
insert Free ops when blob used last time + memory allocation estimator
|
2017-09-05 12:03:04 -07:00 |
|
memonger_test.py
|
insert Free ops when blob used last time + memory allocation estimator
|
2017-09-05 12:03:04 -07:00 |
|
mkl_test_util.py
|
|
|
|
model_device_test.py
|
|
|
|
model_helper.py
|
YellowFin GPU class and Python optimizer
|
2017-08-30 18:32:24 -07:00 |
|
modifier_context.py
|
add base class ModifierContext, rewrite OptimizerContext, add RegularizerContext
|
2017-09-08 11:39:23 -07:00 |
|
mpi_python.cc
|
|
|
|
muji.py
|
|
|
|
muji_test.py
|
|
|
|
net_builder.py
|
Tuning number of parameter servers based on performance estimation job
|
2017-08-30 18:03:59 -07:00 |
|
net_builder_test.py
|
Control flow operators
|
2017-08-28 20:04:43 -07:00 |
|
net_drawer.py
|
|
|
|
net_printer.py
|
net_printer.to_string() accepts NetDef
|
2017-08-01 10:17:29 -07:00 |
|
net_printer_test.py
|
|
|
|
optimizer.py
|
Decayed adagrad
|
2017-09-15 00:35:21 -07:00 |
|
optimizer_context.py
|
add base class ModifierContext, rewrite OptimizerContext, add RegularizerContext
|
2017-09-08 11:39:23 -07:00 |
|
optimizer_test.py
|
Disabled test for equivalency between Caffe2's and Numpy's YellowFin
|
2017-09-06 13:47:45 -07:00 |
|
optimizer_test_util.py
|
Added support for scaling learning rate of Caffe2 optimizers during training
|
2017-08-25 19:04:47 -07:00 |
|
parallel_workers.py
|
Caffe2 [easy]: Better exception logging in parallel_workers/data_workers
|
2017-08-10 15:27:19 -07:00 |
|
parallel_workers_test.py
|
Caffe2: Refactor the core logic from data_workers.py into parallel_workers.py
|
2017-08-07 10:14:08 -07:00 |
|
parallelize_gpu_bmuf_distributed_test.py
|
AddBlobsSync()
|
2017-09-12 10:33:22 -07:00 |
|
pipeline.py
|
|
|
|
pipeline_test.py
|
Ability to dequeue and concat multiple records in a single QueueDequeue op
|
2017-08-31 10:48:59 -07:00 |
|
predictor_constants.py
|
|
|
|
pybind_state.cc
|
Functional C2
|
2017-09-13 15:37:28 -07:00 |
|
pybind_state.h
|
|
|
|
pybind_state_gpu.cc
|
workspace_gpu: Get{CUDAVersion,DeviceProperties}
|
2017-09-17 20:01:34 -07:00 |
|
pybind_state_mkl.cc
|
MKL code move
|
2017-07-26 20:21:55 -07:00 |
|
python_op_test.py
|
|
|
|
queue_util.py
|
Ability to dequeue and concat multiple records in a single QueueDequeue op
|
2017-08-31 10:48:59 -07:00 |
|
record_queue.py
|
|
|
|
recurrent.py
|
make rnn executor figure out recurrent mappings from links
|
2017-09-09 16:19:26 -07:00 |
|
regularizer.py
|
add regulariztion in caffe2 and dper
|
2017-09-08 11:39:22 -07:00 |
|
regularizer_context.py
|
add base class ModifierContext, rewrite OptimizerContext, add RegularizerContext
|
2017-09-08 11:39:23 -07:00 |
|
rnn_cell.py
|
remove in-place Dropout from rnn_cell (bug in PR-1185)
|
2017-09-15 14:03:33 -07:00 |
|
schema.py
|
logging the blob that has type error
|
2017-08-23 21:21:27 -07:00 |
|
schema_test.py
|
Return empty Struct when get_field has empty input
|
2017-08-01 19:49:47 -07:00 |
|
scope.py
|
|
|
|
scope_test.py
|
|
|
|
session.py
|
Adds the master setup plan to the model exporter.
|
2017-08-25 16:01:24 -07:00 |
|
session_test.py
|
|
|
|
sparse_to_dense_mask_test.py
|
Add more enforces to SparseToDenseMask operator.
|
2017-09-02 02:16:24 -07:00 |
|
task.py
|
Support session in distributed realtime trainer
|
2017-08-16 10:28:55 -07:00 |
|
test_util.py
|
Clear the operator default engines before running operator tests
|
2017-08-29 17:47:20 -07:00 |
|
text_file_reader.py
|
|
|
|
timeout_guard.py
|
Revert D5655753: [Caffe2] better straggler exit procedure
|
2017-08-25 14:23:09 -07:00 |
|
toy_regression_test.py
|
|
|
|
tt_core.py
|
|
|
|
tt_core_test.py
|
|
|
|
utils.py
|
Update proto definition
|
2017-08-22 19:01:18 -07:00 |
|
visualize.py
|
|
|
|
workspace.py
|
workspace_gpu: Get{CUDAVersion,DeviceProperties}
|
2017-09-17 20:01:34 -07:00 |
|
workspace_test.py
|
ApplyTransformIfFaster
|
2017-08-17 15:36:51 -07:00 |
ebeaecbfa3