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

diff --git a/caffe2/operators/onnxifi_op.cc b/caffe2/operators/onnxifi_op.cc
index a375f85..903f829 100644 (file)
--- a/caffe2/operators/onnxifi_op.cc
+++ b/caffe2/operators/onnxifi_op.cc
@@ -138,7 +138,7 @@ void BlobToTensorDescriptor(
 
 template <>
 std::vector<onnxTensorDescriptorV1>
-OnnxifiOp<float, CPUContext>::buildInitializationList(
+OnnxifiOp<CPUContext>::buildInitializationList(
     Workspace* ws,
     const std::vector<std::string>& initializers,
     std::vector<std::string>* weight_names,
@@ -167,7 +167,7 @@ OnnxifiOp<float, CPUContext>::buildInitializationList(
 }
 
 template <>
-std::vector<int> OnnxifiOp<float, CPUContext>::extractOutputBatchSizes() const {
+std::vector<int> OnnxifiOp<CPUContext>::extractOutputBatchSizes() const {
   CAFFE_ENFORCE_EQ(
       input_shapes_.size(),
       InputSize(),
@@ -218,7 +218,7 @@ std::vector<int> OnnxifiOp<float, CPUContext>::extractOutputBatchSizes() const {
 }
 
 template <>
-void OnnxifiOp<float, CPUContext>::maybeAdjustOutputBatchSizes(
+void OnnxifiOp<CPUContext>::maybeAdjustOutputBatchSizes(
     const std::vector<int>& 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<float, CPUContext>::maybeAdjustOutputBatchSizes(
 }
 
 template <>
-bool OnnxifiOp<float, CPUContext>::RunOnDevice() {
+bool OnnxifiOp<CPUContext>::RunOnDevice() {
   CAFFE_ENFORCE_EQ(input_desc_.size(), InputSize());
   input_shapes_.clear();
   for (unsigned i = 0U; i < InputSize(); ++i) {
@@ -342,7 +342,7 @@ bool OnnxifiOp<float, CPUContext>::RunOnDevice() {
   return true;
 }
 
-REGISTER_CPU_OPERATOR(Onnxifi, OnnxifiOp<float, CPUContext>);
+REGISTER_CPU_OPERATOR(Onnxifi, OnnxifiOp<CPUContext>);
 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 (file)
--- a/caffe2/operators/onnxifi_op.h
+++ b/caffe2/operators/onnxifi_op.h
@@ -13,7 +13,7 @@
 
 namespace caffe2 {
 
-template <typename T, typename Context>
+template <typename Context>
 class OnnxifiOp final : public Operator<Context> {
   struct TensorInfo {
     TensorInfo() {}