aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/plugin/example/executable.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/plugin/example/executable.h')
-rw-r--r--tensorflow/compiler/plugin/example/executable.h18
1 files changed, 8 insertions, 10 deletions
diff --git a/tensorflow/compiler/plugin/example/executable.h b/tensorflow/compiler/plugin/example/executable.h
index 0d3dc3f682..99d2da77dd 100644
--- a/tensorflow/compiler/plugin/example/executable.h
+++ b/tensorflow/compiler/plugin/example/executable.h
@@ -41,25 +41,23 @@ class ExampleExecutable : public Executable {
std::unique_ptr<HloModuleConfig> module_config);
~ExampleExecutable() override;
-
StatusOr<perftools::gputools::DeviceMemoryBase> ExecuteOnStream(
- const ServiceExecutableRunOptions* run_options,
- tensorflow::gtl::ArraySlice<perftools::gputools::DeviceMemoryBase>
+ const ServiceExecutableRunOptions* run_options,
+ tensorflow::gtl::ArraySlice<perftools::gputools::DeviceMemoryBase>
arguments,
- HloExecutionProfile* hlo_execution_profile) override;
+ 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;
+ 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>
+ const ServiceExecutableRunOptions* run_options,
+ tensorflow::gtl::ArraySlice<perftools::gputools::DeviceMemoryBase>
arguments) override;
private:
-
TF_DISALLOW_COPY_AND_ASSIGN(ExampleExecutable);
};