pytorch/caffe2/python
Junjie Bai 90ca470d70 Standardize operator argument "is_test"
Summary:
Also add the ability to mark an argument as required.

Added a string constant `OpSchema::Arg_IsTest` for `is_test` arg.
If users define the `is_test` argument with `ArgIsTest(...)`, then it automatically becomes required argument, in the meanwhile user can still use `Arg("is_test", ...)` to define an optional `is_test` argument.

Reviewed By: akyrola

Differential Revision: D5812391

fbshipit-source-id: eaaba50d027813a8012389edc6c459de23c3c728
2017-09-13 14:35:27 -07:00
..
docs Standardize operator argument "is_test" 2017-09-13 14:35:27 -07:00
examples Create Gloo common world through MPI rendezvous 2017-09-08 17:18:47 -07:00
helpers fp16: BatchMatMul 2017-09-13 14:35:25 -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 Enable target dialect token in inference. 2017-09-12 17:22:18 -07:00
operator_test Standardize operator argument "is_test" 2017-09-13 14:35:27 -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 fp16: BatchMatMul 2017-09-13 14:35:25 -07:00
benchmark_generator.py A benchmark generator for individual ops 2017-08-31 17:33:21 -07:00
binarysize.py
brew.py fp16: BatchMatMul 2017-09-13 14:35:25 -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
caffe_translator_test.py
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 tests: use assertRaises, not expectedFail 2017-09-13 11:39:35 -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 fix timeouts in CloneOrCreateCommonWorld 2017-09-12 13:09:05 -07:00
data_parallel_model_test.py tests: use assertRaises, not expectedFail 2017-09-13 11:39:35 -07:00
data_workers.py
data_workers_test.py
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
empty.so
experiment_util.py
extension_loader.py
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 a range operator similar to np.arange 2017-08-18 14:45:56 -07:00
layer_model_helper.py
layer_model_instantiator.py
layer_parameter_sharing_test.py
layer_test_util.py
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_test.py
optimizer.py YellowFin GPU class and Python optimizer 2017-08-30 18:32:24 -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
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 Standardize operator argument "is_test" 2017-09-13 14:35:27 -07:00
pybind_state.h
pybind_state_gpu.cc
pybind_state_mkl.cc
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 Use brew in rnn_cell.py 2017-09-13 12:02:57 -07:00
schema.py logging the blob that has type error 2017-08-23 21:21:27 -07:00
schema_test.py
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 ApplyTransformIfFaster 2017-08-17 15:36:51 -07:00
workspace_test.py ApplyTransformIfFaster 2017-08-17 15:36:51 -07:00