Function AssignExpressionToTensor

tensorflow/compiler/tf2xla/xla_op_kernel.cc:62–69

Assigns an XlaExpression to a tensor on an XLA compilation device.

// Assigns an XlaExpression to a tensor on an XLA compilation device.
static void AssignExpressionToTensor(Tensor* tensor,
                                     const XlaExpression& value) {
  const XlaExpression* expression =
      reinterpret_cast<const XlaExpression*>(tensor->tensor_data().data());
  CHECK(expression->kind() == XlaExpression::Kind::kInvalid)
      << expression->HumanString();
  *const_cast<XlaExpression*>(expression) = value;
}

const XlaExpression& XlaOpKernelContext::InputExpression(int index) {
  return *CastExpressionFromTensor(context_->input(index));