diff options
Diffstat (limited to 'tensorflow/compiler/jit/xla_device_ops.cc')
-rw-r--r-- | tensorflow/compiler/jit/xla_device_ops.cc | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/tensorflow/compiler/jit/xla_device_ops.cc b/tensorflow/compiler/jit/xla_device_ops.cc new file mode 100644 index 0000000000..74c314c8ed --- /dev/null +++ b/tensorflow/compiler/jit/xla_device_ops.cc @@ -0,0 +1,36 @@ +/* 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. +==============================================================================*/ + +#include "tensorflow/compiler/jit/xla_device_ops.h" + +#include "tensorflow/compiler/jit/xla_device_context.h" + +namespace tensorflow { + +void XlaDeviceAssignOp::Copy(OpKernelContext* context, Tensor* lhs, + const Tensor& rhs) { + std::shared_ptr<xla::GlobalData> gd = + XlaTransferManager::GetTensorGlobalData(rhs); + XlaTransferManager::SetTensorGlobalData(std::move(gd), lhs); +} + +XlaDeviceDummyOp::XlaDeviceDummyOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} + +void XlaDeviceDummyOp::Compute(OpKernelContext* ctx) { + LOG(FATAL) << "Attempted to execute Op " << name() << "type " << type_string() + << " on an XLA device. This should never happen."; +} + +} // namespace tensorflow |