diff options
| author | Christian Puhrsch <cpuhrsch@fb.com> | 2018-09-22 18:07:38 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-09-22 18:11:04 -0700 |
| commit | a6630e25afb49642cf517ee948f5a0c06a9cba6d (patch) | |
| tree | 9b8e21397d409ae522d8d50c7bfa78f22f48ae52 /caffe2/opt | |
| parent | 5d0f1c3c8ff4cb14bdf554b447d6851877d710ac (diff) | |
| download | pytorch-a6630e25afb49642cf517ee948f5a0c06a9cba6d.tar.gz pytorch-a6630e25afb49642cf517ee948f5a0c06a9cba6d.tar.bz2 pytorch-a6630e25afb49642cf517ee948f5a0c06a9cba6d.zip | |
Remove many caffe2::TIndex and replace them with int64_t (#11943)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11943
See title
Reviewed By: ezyang
Differential Revision: D9992645
fbshipit-source-id: e8f80d6ea762971513e5e8072975ceea53e1f11a
Diffstat (limited to 'caffe2/opt')
| -rw-r--r-- | caffe2/opt/onnxifi_transformer.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/caffe2/opt/onnxifi_transformer.cc b/caffe2/opt/onnxifi_transformer.cc index 2f48ece4fd..ce79df56ec 100644 --- a/caffe2/opt/onnxifi_transformer.cc +++ b/caffe2/opt/onnxifi_transformer.cc @@ -174,7 +174,7 @@ NetDef OnnxifiTransformer::SubnetToOnnxifiOp( // Feed into workspace as CPU Tensors auto* blob = ws->CreateBlob(t.name()); auto* cpu_tensor = blob->GetMutableTensor(CPU); - std::vector<TIndex> dims; + std::vector<int64_t> dims; for(const auto& d : t.dims()) { dims.push_back(d); } |