pytorch/caffe2/opt
Yinghai Lu 7e6a55e417 Add DimType info in dumped debug nets (#26589)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26589

Just for better debugging purpose.

Test Plan: Dump the net and check the dim type info is in the pb_txt.

Reviewed By: dreamingleo

Differential Revision: D17505931

fbshipit-source-id: ceba4c3849eb271c22227fa07a05d5bcb07344a5
2019-09-20 21:46:19 -07:00
..
custom Enable loading int8 prepacked models in PredictorContainer 2019-09-06 02:53:52 -07:00
nql BlackBoxPredictor OSS: open-source NQL and custom transforms (#22877) 2019-07-19 14:37:56 -07:00
annotations.cc
annotations.h
backend_cutting.cc Fuse to individual operators to GatherFuse8BitRowwiseQuantFloatMulLengthElim (#25519) 2019-09-03 19:08:49 -07:00
backend_cutting.h Fuse to individual operators to GatherFuse8BitRowwiseQuantFloatMulLengthElim (#25519) 2019-09-03 19:08:49 -07:00
backend_cutting_test.cc
backend_transformer_base.cc Add DimType info in dumped debug nets (#26589) 2019-09-20 21:46:19 -07:00
backend_transformer_base.h C2/glow: assign net_pos to a net before applying onnxifi_blacklist_ops (#24262) 2019-08-14 10:39:15 -07:00
bound_shape_inference_test.cc bucketize op shape inference (#22716) 2019-07-11 12:44:29 -07:00
bound_shape_inferencer.cc Add DimType info in dumped debug nets (#26589) 2019-09-20 21:46:19 -07:00
bound_shape_inferencer.h Change bound shape inference for LengthsRangeFill & GatherRanges, add more tests (#20610) 2019-05-28 11:33:51 -07:00
CMakeLists.txt
converter.cc
converter.h
converter_nomigraph_test.cc
dead_code_elim.cc
dead_code_elim_test.cc
device.cc
device.h
device_test.cc
distributed.cc
distributed.h
distributed_converter.cc
distributed_test.cc
fusion.cc
fusion.h
mobile.cc
mobile.h
mobile_test.cc
onnx_convert.h
onnxifi_op.cc Support lowering of fp16 weights 2019-08-23 16:06:15 -07:00
onnxifi_op.h Fix more warnings 2019-08-14 17:47:54 -07:00
onnxifi_transformer.cc cleanup warnings 2019-08-12 16:12:30 -07:00
onnxifi_transformer.h Reuse common options between ONNXIFI and TVM transformations (#21163) 2019-05-31 17:01:36 -07:00
optimize_ideep.cc upgrade mkldnn-bridge (#20569) 2019-06-11 00:47:11 -07:00
optimize_ideep.h
optimizer.cc
optimizer.h
passes.cc
passes.h
shape_info.cc Fbgemm fp16 tensor support (#23101) 2019-07-19 17:08:03 -07:00
shape_info.h Remove useless code from shape info 2019-08-01 11:47:01 -07:00
tvm_transformer.cc Set caffe2_tvm_min_ops to 8 (#23893) 2019-08-06 14:48:45 -07:00
tvm_transformer.h format changes 2019-08-14 01:08:31 -07:00