pytorch/caffe2/python
François Garillot b6adecdeee correct schema.Scalar's shape for a shape argument of 1 (#6493)
The schema.Scalar class makes pretty strict assumptions (via its docstring)
on the spec of the shape of its underlying object. Because of idiosyncracies
of numpy indexing and the use of np.dtype, those assumptions are broken on an
 edge case (dtype = (scalar_type, 1)). This corrects the behavior of this
edge case to conform to the spec.
2018-05-07 18:58:11 -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 [Caffe2] Clean up ideep integration (#6881) 2018-04-24 18:32:35 -07:00
layers [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
mint Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mkl [Caffe2] Clean up ideep integration (#6881) 2018-04-24 18:32:35 -07:00
modeling Update from Facebook (#6692) 2018-04-17 23:36:40 -07:00
models Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
onnx Refactor rnn export (#7263) 2018-05-04 14:00:09 -07:00
operator_test Fix lint errors (#7247) 2018-05-03 12:17:23 -07:00
predictor Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rnn Fix a typo 2018-03-30 21:00:44 -07:00
test Fix occasional test timeouts 2018-01-19 20:08:58 -08:00
trt [Caffe2] Follow-up of onnx-trt API change (#7076) 2018-04-28 23:07:15 -07:00
__init__.py Experimental support for setup.py develop mode install 2018-02-12 23:36:18 -08:00
_import_c_extension.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
allcompare_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
attention.py Remove Apache headers from source. 2018-03-27 13:10:18 -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 Remove Apache headers from source. 2018-03-27 13:10:18 -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 Remove Apache headers from source. 2018-03-27 13:10:18 -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] Clean up ideep integration (#6881) 2018-04-24 18:32:35 -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 Remove Apache headers from source. 2018-03-27 13:10:18 -07: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 Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
core_test.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
core.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
crf.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
data_parallel_model_test.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
data_parallel_model.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
data_workers_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
data_workers.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
dataio_test.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
dataio.py Update from Facebook (#6692) 2018-04-17 23:36:40 -07:00
dataset.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
db_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
device_checker.py Remove Apache headers from source. 2018-03-27 13:10:18 -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 Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
gradient_checker.py Remove Apache headers from source. 2018-03-27 13:10:18 -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 Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
hypothesis_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -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 [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -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 [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -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 Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
muji.py Remove Apache headers from source. 2018-03-27 13:10:18 -07: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
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 Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
optimizer.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
parallel_workers_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
parallel_workers.py Remove Apache headers from source. 2018-03-27 13:10:18 -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 Remove Apache headers from source. 2018-03-27 13:10:18 -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_ideep.cc Move ideep ops from caffe2/contrib/ideep to caffe2/ideep (#7112) 2018-04-30 21:10:46 -07:00
pybind_state_int8.cc [C2 Int8][C2 Core]fetch int8 blob 2018-03-30 21:00:44 -07:00
pybind_state_mkl.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
pybind_state.cc [caffe2][nomnigraph] Add maxpool sink transform (#7207) 2018-05-07 14:52:10 -07:00
pybind_state.h [Caffe2] Workspace centric API for TensorRT transformation (#6678) 2018-04-17 21:23:27 -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 Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
regularizer_test.py add dense regularization 2018-01-08 13:03:17 -08:00
regularizer.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rnn_cell.py Fix typo in comment (#7183) 2018-05-02 11:58:30 -07:00
schema_test.py correct schema.Scalar's shape for a shape argument of 1 (#6493) 2018-05-07 18:58:11 -07:00
schema.py correct schema.Scalar's shape for a shape argument of 1 (#6493) 2018-05-07 18:58:11 -07:00
scope_test.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
scope.py Remove Apache headers from source. 2018-03-27 13:10:18 -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 [caffe2][nomnigraph] Add maxpool sink transform (#7207) 2018-05-07 14:52:10 -07:00
transformations.py [caffe2][nomnigraph] Add maxpool sink transform (#7207) 2018-05-07 14:52:10 -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 [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -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 change print to logger.warning in operator traceback code (#6216) 2018-04-03 08:01:25 -07:00