From c746bb741c3fbf7cafb6317ace1d9427278f9dbc Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 10 Jan 2022 07:35:41 -0500 Subject: [PATCH] remove `api_cc/include/custom_op.h` This file is duplicate of `op/custom_op.h`. It's not used by files outside the `op` directory. --- source/api_cc/include/custom_op.h | 34 ------------------------------- 1 file changed, 34 deletions(-) delete mode 100644 source/api_cc/include/custom_op.h diff --git a/source/api_cc/include/custom_op.h b/source/api_cc/include/custom_op.h deleted file mode 100644 index bd27821fb3..0000000000 --- a/source/api_cc/include/custom_op.h +++ /dev/null @@ -1,34 +0,0 @@ -#include -#include -#include - -#include "tensorflow/core/platform/env.h" -#include "tensorflow/core/framework/op.h" -#include "tensorflow/core/public/session.h" -#include "tensorflow/core/public/version.h" -#include "tensorflow/core/framework/op_kernel.h" -#include "tensorflow/core/framework/shape_inference.h" -#include -#include - -using namespace tensorflow; -using CPUDevice = Eigen::ThreadPoolDevice; -using GPUDevice = Eigen::GpuDevice; - -// functions used in custom ops -struct DeviceFunctor { - void operator()( - std::string& device, - const CPUDevice& d) - { - device = "CPU"; - } - #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM - void operator()( - std::string& device, - const GPUDevice& d) - { - device = "GPU"; - } - #endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM -}; \ No newline at end of file