diff options
Diffstat (limited to 'binaries/speed_benchmark.cc')
| -rw-r--r-- | binaries/speed_benchmark.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/binaries/speed_benchmark.cc b/binaries/speed_benchmark.cc index 5914e3f58b..fd502cf3c0 100644 --- a/binaries/speed_benchmark.cc +++ b/binaries/speed_benchmark.cc @@ -137,7 +137,7 @@ int main(int argc, char** argv) { if (blob == nullptr) { blob = workspace->CreateBlob(input_names[i]); } - caffe2::TensorCPU* tensor = blob->GetMutableTensor(caffe2::CPU); + caffe2::TensorCPU* tensor = BlobGetMutableTensor(blob, caffe2::CPU); CHECK_NOTNULL(tensor); tensor->Resize(input_dims); if (input_type_list[i] == "uint8_t") { |