diff options
| author | Mikhail Zolotukhin <mvz@fb.com> | 2019-04-21 20:28:15 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-21 20:31:09 -0700 |
| commit | 868933a467abf82d6b6673d1f1a73f87e0d6e280 (patch) | |
| tree | e1d4db05443fa1a6a5275eb94d694c5164c2c382 | |
| parent | 1bd5f2c1817cf4304527d35b616811c6c14c9763 (diff) | |
| download | pytorch-868933a467abf82d6b6673d1f1a73f87e0d6e280.tar.gz pytorch-868933a467abf82d6b6673d1f1a73f87e0d6e280.tar.bz2 pytorch-868933a467abf82d6b6673d1f1a73f87e0d6e280.zip | |
Fix clang-format. (#19550)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19550
ghimport-source-id: 980d96762426d3e97c26839edbaf107a3fc18b2f
Differential Revision: D15028055
Pulled By: ZolotukhinM
fbshipit-source-id: a50a0aaa74d0f1b9249ad79ab80e4b7747c3bffc
| -rw-r--r-- | torch/csrc/jit/init.cpp | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/torch/csrc/jit/init.cpp b/torch/csrc/jit/init.cpp index 3ad5876d69..87014abf22 100644 --- a/torch/csrc/jit/init.cpp +++ b/torch/csrc/jit/init.cpp @@ -22,10 +22,10 @@ #include <torch/csrc/jit/passes/loop_unrolling.h> #include <torch/csrc/jit/passes/lower_tuples.h> #include <torch/csrc/jit/passes/onnx.h> +#include <torch/csrc/jit/passes/onnx/constant_fold.h> #include <torch/csrc/jit/passes/onnx/fixup_onnx_loop.h> #include <torch/csrc/jit/passes/onnx/peephole.h> #include <torch/csrc/jit/passes/onnx/prepare_division_for_onnx.h> -#include <torch/csrc/jit/passes/onnx/constant_fold.h> #include <torch/csrc/jit/passes/peephole.h> #include <torch/csrc/jit/passes/quantization.h> #include <torch/csrc/jit/passes/remove_expands.h> @@ -103,12 +103,13 @@ void initJITBindings(PyObject* module) { .def("_jit_pass_lower_all_tuples", LowerAllTuples) .def("_jit_pass_onnx_peephole", PeepholeOptimizeONNX) .def( - "_jit_pass_onnx_constant_fold", - [](std::shared_ptr<Graph>& graph, - std::map<std::string, at::Tensor>& paramsDict) { - ConstantFoldONNX(graph->block(), paramsDict); // overload resolution - return paramsDict; - }, pybind11::return_value_policy::move) + "_jit_pass_onnx_constant_fold", + [](std::shared_ptr<Graph>& graph, + std::map<std::string, at::Tensor>& paramsDict) { + ConstantFoldONNX(graph->block(), paramsDict); // overload resolution + return paramsDict; + }, + pybind11::return_value_policy::move) .def("_jit_pass_fuse", FuseGraph) .def( "_jit_pass_dce", |