diff options
Diffstat (limited to 'tensorflow/compiler/plugin/executor/executable.h')
-rw-r--r-- | tensorflow/compiler/plugin/executor/executable.h | 65 |
1 files changed, 0 insertions, 65 deletions
diff --git a/tensorflow/compiler/plugin/executor/executable.h b/tensorflow/compiler/plugin/executor/executable.h deleted file mode 100644 index ba3d4da21d..0000000000 --- a/tensorflow/compiler/plugin/executor/executable.h +++ /dev/null @@ -1,65 +0,0 @@ -/* Copyright 2017 The TensorFlow Authors. All Rights Reserved. - -Licensed under the Apache License, Version 2.0 (the "License"); -you may not use this file except in compliance with the License. -You may obtain a copy of the License at - - http://www.apache.org/licenses/LICENSE-2.0 - -Unless required by applicable law or agreed to in writing, software -distributed under the License is distributed on an "AS IS" BASIS, -WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -See the License for the specific language governing permissions and -limitations under the License. -==============================================================================*/ - -#ifndef TENSORFLOW_COMPILER_EXECUTOR_DRIVER_EXECUTOR_EXECUTABLE_H_ -#define TENSORFLOW_COMPILER_EXECUTOR_DRIVER_EXECUTOR_EXECUTABLE_H_ - -#include <cstddef> -#include <memory> -#include <string> -#include <unordered_map> -#include <vector> - -#include "tensorflow/compiler/xla/service/executable.h" -#include "tensorflow/compiler/xla/service/hlo_module.h" -#include "tensorflow/compiler/xla/service/hlo_module_config.h" - -#include "tensorflow/stream_executor/lib/status.h" -#include "tensorflow/stream_executor/lib/statusor.h" - -namespace xla { -namespace executorplugin { - -class ExecutorExecutable : public Executable { - public: - ExecutorExecutable(std::unique_ptr<HloModule> hlo_module); - ~ExecutorExecutable() override; - - StatusOr<perftools::gputools::DeviceMemoryBase> ExecuteOnStream( - const ServiceExecutableRunOptions* run_options, - tensorflow::gtl::ArraySlice<perftools::gputools::DeviceMemoryBase> - arguments, - HloExecutionProfile* hlo_execution_profile) override; - - StatusOr<std::unique_ptr<ShapedBuffer>> ExecuteOnStream( - const ServiceExecutableRunOptions* run_options, - tensorflow::gtl::ArraySlice<const ShapedBuffer*> arguments, - HloExecutionProfile* hlo_execution_profile) override; - - StatusOr<perftools::gputools::DeviceMemoryBase> ExecuteAsyncOnStream( - const ServiceExecutableRunOptions* run_options, - tensorflow::gtl::ArraySlice<perftools::gputools::DeviceMemoryBase> - arguments) override; - - static int64 ShapeSizeBytes(const Shape& shape); - - private: - TF_DISALLOW_COPY_AND_ASSIGN(ExecutorExecutable); -}; - -} // namespace executorplugin -} // namespace xla - -#endif // TENSORFLOW_COMPILER_EXECUTOR_DRIVER_EXECUTOR_EXECUTABLE_H_ |