diff options
Diffstat (limited to 'tensorflow/compiler/plugin/example/executable.h')
-rw-r--r-- | tensorflow/compiler/plugin/example/executable.h | 18 |
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); }; |