text: //==============================================================================
  //
  //  @@-COPYRIGHT-START-@@
  //
  //  Copyright (c) 2020-2022, Qualcomm Innovation Center, Inc. All rights reserved.
  //
  //  Redistribution and use in source and binary forms, with or without
  //  modification, are permitted provided that the following conditions are met:
  //
  //  1. Redistributions of source code must retain the above copyright notice,
  //     this list of conditions and the following disclaimer.
  //
  //  2. Redistributions in binary form must reproduce the above copyright notice,
  //     this list of conditions and the following disclaimer in the documentation
  //     and/or other materials provided with the distribution.
  //
  //  3. Neither the name of the copyright holder nor the names of its contributors
  //     may be used to endorse or promote products derived from this software
  //     without specific prior written permission.
  //
  //  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  //  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  //  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  //  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
  //  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  //  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  //  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  //  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  //  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  //  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  //  POSSIBILITY OF SUCH DAMAGE.
  //
  //  SPDX-License-Identifier: BSD-3-Clause
  //
  //  @@-COPYRIGHT-END-@@
  //
  //==============================================================================
  
  #ifndef AIMET_OP_UTILS_H
  #define AIMET_OP_UTILS_H
  
  #include "tensorflow/core/framework/op.h"
  #include "tensorflow/core/framework/op_kernel.h"
  #include "tensorflow/core/framework/shape_inference.h"
  #include "tensorflow/core/platform/logging.h"
  
  #include <DlQuantization/TensorQuantizerOpFacade.h>
  
  // Declarations of the functors to be used
  // to invoke Quantization APIs.
  using namespace tensorflow;
  
  using CPUDevice = Eigen::ThreadPoolDevice;
  using GPUDevice = Eigen::GpuDevice;
  
  template <typename T>
  void copyInputTensorsToOutputTensors(const CPUDevice& d, const T* inTensor, size_t count, T* outTensor);
  
  template <typename T>
  T copyLiteralToHost(const CPUDevice& d, const T* deviceValue);
  
  template <typename T>
  void copyInputTensorsToOutputTensors(const GPUDevice& d, const T* inTensor, size_t count, T* outTensor);
  
  template <typename T>
  T copyLiteralToHost(const GPUDevice& d, const T* deviceValue);
  
  void sliceTensorAlongLastDim(const CPUDevice& d, Tensor slicedTensor, const Tensor& tensorToSlice, int channel);
  
  void sliceTensorAlongLastDim(const GPUDevice& d, Tensor slicedTensor, const Tensor& tensorToSlice, int channel);
  
  void sliceAndStoreTensor(const CPUDevice& d, Tensor* slicedTensor, Tensor tensorToSlice, int channel);
  
  void sliceAndStoreTensor(const GPUDevice& d, Tensor* slicedTensor, Tensor tensorToSlice, int channel);
  
  #endif   // AIMET_OP_UTILS_H
  