From 5fa1aad67077af90f1f0940ff9c7cebb03a04e3c Mon Sep 17 00:00:00 2001 From: Yinghai Lu Date: Wed, 17 Apr 2019 16:40:58 -0700 Subject: [PATCH] Remove unused template parameter in OnnxifiOp (#19362) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19362 `float` type is never used in OnnxifiOp.... Reviewed By: bddppq Differential Revision: D14977970 fbshipit-source-id: 8fee02659dbe408e5a3e0ff95d74c04836c5c281 --- caffe2/operators/onnxifi_op.cc | 10 +++++----- caffe2/operators/onnxifi_op.h | 2 +- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/caffe2/operators/onnxifi_op.cc b/caffe2/operators/onnxifi_op.cc index a375f85..903f829 100644 --- a/caffe2/operators/onnxifi_op.cc +++ b/caffe2/operators/onnxifi_op.cc @@ -138,7 +138,7 @@ void BlobToTensorDescriptor( template <> std::vector -OnnxifiOp::buildInitializationList( +OnnxifiOp::buildInitializationList( Workspace* ws, const std::vector& initializers, std::vector* weight_names, @@ -167,7 +167,7 @@ OnnxifiOp::buildInitializationList( } template <> -std::vector OnnxifiOp::extractOutputBatchSizes() const { +std::vector OnnxifiOp::extractOutputBatchSizes() const { CAFFE_ENFORCE_EQ( input_shapes_.size(), InputSize(), @@ -218,7 +218,7 @@ std::vector OnnxifiOp::extractOutputBatchSizes() const { } template <> -void OnnxifiOp::maybeAdjustOutputBatchSizes( +void OnnxifiOp::maybeAdjustOutputBatchSizes( const std::vector& real_output_batch_sizes) { CAFFE_ENFORCE_EQ(real_output_batch_sizes.size(), output_shapes_.size()); for (int i = 0; i < real_output_batch_sizes.size(); ++i) { @@ -231,7 +231,7 @@ void OnnxifiOp::maybeAdjustOutputBatchSizes( } template <> -bool OnnxifiOp::RunOnDevice() { +bool OnnxifiOp::RunOnDevice() { CAFFE_ENFORCE_EQ(input_desc_.size(), InputSize()); input_shapes_.clear(); for (unsigned i = 0U; i < InputSize(); ++i) { @@ -342,7 +342,7 @@ bool OnnxifiOp::RunOnDevice() { return true; } -REGISTER_CPU_OPERATOR(Onnxifi, OnnxifiOp); +REGISTER_CPU_OPERATOR(Onnxifi, OnnxifiOp); OPERATOR_SCHEMA(Onnxifi) .NumInputs(0, INT_MAX) .NumOutputs(0, INT_MAX) diff --git a/caffe2/operators/onnxifi_op.h b/caffe2/operators/onnxifi_op.h index a8d0bf4..c92dde8 100644 --- a/caffe2/operators/onnxifi_op.h +++ b/caffe2/operators/onnxifi_op.h @@ -13,7 +13,7 @@ namespace caffe2 { -template +template class OnnxifiOp final : public Operator { struct TensorInfo { TensorInfo() {} -- 2.7.4