aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/plugin/executor/transfer_manager.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/plugin/executor/transfer_manager.h')
-rw-r--r--tensorflow/compiler/plugin/executor/transfer_manager.h77
1 files changed, 0 insertions, 77 deletions
diff --git a/tensorflow/compiler/plugin/executor/transfer_manager.h b/tensorflow/compiler/plugin/executor/transfer_manager.h
deleted file mode 100644
index 7a42e5a2d7..0000000000
--- a/tensorflow/compiler/plugin/executor/transfer_manager.h
+++ /dev/null
@@ -1,77 +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_TRANSFER_MANAGER_H_
-#define TENSORFLOW_COMPILER_EXECUTOR_DRIVER_EXECUTOR_TRANSFER_MANAGER_H_
-
-#include "tensorflow/compiler/xla/service/transfer_manager.h"
-#include "tensorflow/compiler/xla/statusor.h"
-#include "tensorflow/compiler/xla/xla_data.pb.h"
-#include "tensorflow/core/platform/macros.h"
-#include "tensorflow/core/platform/stream_executor_no_cuda.h"
-#include "tensorflow/core/platform/types.h"
-
-#include <vector>
-
-namespace se = ::perftools::gputools;
-
-namespace xla {
-namespace executorplugin {
-
-class ExecutorTransferManager : public TransferManager {
- public:
- ExecutorTransferManager();
-
- ~ExecutorTransferManager() override {}
-
- se::Platform::Id PlatformId() const override;
-
- StatusOr<std::vector<se::DeviceMemoryBase>> ShallowCopyTupleFromDevice(
- se::StreamExecutor* executor, const se::DeviceMemoryBase& source,
- const Shape& shape) override;
-
- Status TransferLiteralFromDevice(se::StreamExecutor* executor,
- const se::DeviceMemoryBase& source,
- const Shape& device_shape,
- const Shape& literal_shape,
- Literal* literal) override;
-
- Status TransferLiteralToDevice(se::StreamExecutor* executor,
- const Literal& literal,
- se::DeviceMemoryBase* destination) override;
-
- Status TransferLiteralToInfeed(se::StreamExecutor* executor,
- const Literal& literal) override;
-
- Status TransferBufferToInfeed(se::StreamExecutor* executor,
- int64 size, const void* source) override;
-
- Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
- const Shape& literal_shape,
- Literal* literal) override;
-
- Status ResetDevices(
- tensorflow::gtl::ArraySlice<se::StreamExecutor*> executors) override;
-
- int64 GetByteSizeRequirement(const Shape& shape) override;
-
- private:
- TF_DISALLOW_COPY_AND_ASSIGN(ExecutorTransferManager);
-};
-
-} // namespace executorplugin
-} // namespace xla
-
-#endif // TENSORFLOW_COMPILER_EXECUTOR_DRIVER_EXECUTOR_TRANSFER_MANAGER_H_