diff options
| author |  Justin Lebar <jlebar@google.com> | 2018-04-23 17:16:55 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-23 17:19:22 -0700 |
| commit | 80fc661853f9a0844faf95eb68438dc85a5879e3 (patch) | |
| tree | e7328a367c4f4db472063f96c983b70e63aa3f26 /tensorflow/compiler/aot | |
| parent | ecd837fd0ab69cf54d920eae3b1c73602be6c626 (diff) | |
Use tensorflow::se instead of perftools::gputools for StreamExecutor.
PiperOrigin-RevId: 194010749
Diffstat (limited to 'tensorflow/compiler/aot')
| -rw-r--r-- | tensorflow/compiler/aot/compile.cc | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/compiler/aot/compile.cc b/tensorflow/compiler/aot/compile.cc index 7c83387881..e17a7c4bf6 100644 --- a/tensorflow/compiler/aot/compile.cc +++ b/tensorflow/compiler/aot/compile.cc @@ -88,9 +88,8 @@ Status CompileGraph(const GraphDef& graph_def, const tf2xla::Config& config, // Converts the graph into an XLA computation, and compiles the // computation. // TODO(toddw): Should we let the user pick the XLA cpu vs. gpu client? - namespace gpu = perftools::gputools; - gpu::Platform* cpu_platform = - gpu::MultiPlatformManager::PlatformWithName("Host").ValueOrDie(); + se::Platform* cpu_platform = + se::MultiPlatformManager::PlatformWithName("Host").ValueOrDie(); xla::CompileOnlyClient* client = xla::ClientLibrary::GetOrCreateCompileOnlyClient(cpu_platform) .ValueOrDie(); |