TensorFlow：如何获取指向ConstTensor数据内容的指针？

我的团队正在为TensorFlow开发一个新的后端。通常，tensorflow opkernels作为参数“Tensor”类型传递，这些参数使用我们的体系结构分配的内存：

void ComputeAsync(OpKernelContext* context, DoneCallback done) override {
  // Grab the input tensors
  const Tensor& A = context->input(0);
  const Tensor& B = context->input(1);

  // ...input validation...

  const our::Memory_Type *bA = static_cast<const our::Memory_Type *>(DMAHelper::base(&A));
  const our::Memory_Type *bB = static_cast<const our::Memory_Type *>(DMAHelper::base(&B));

  // ...additional preconditioning...

  // Create an output tensor
  Tensor *C = NULL;
  OP_REQUIRES_OK(context, context->allocate_output(0, out_shape, &C));
  our::Memory_Type *bC = static_cast<kpi::KPI_RMR_Mem*>(DMAHelper::base(C));

  //and run it
  our_impl(bA, bB, bC, done);
}

// in0, in1, and output are all tensorflow::Tensor types, but ConstTensor is an Eigen type
typename TTypes<Type, 2>::ConstTensor in0_data =
    in0.flat_inner_dims<Type>();
typename TTypes<Type, 2>::ConstTensor in1_data =
    in1.flat_inner_dims<Type>();
typename TTypes<Type, 2>::Tensor output_data =
    output->flat_inner_dims<Type>();

const our::Memory_Type *in0_arg = static_cast<const our::Memory_Type *>(DMAHelper::base(&in0));
const our::Memory_Type *in1_arg = static_cast<const our::Memory_Type *>(DMAHelper::base(&in1));
const our::Memory_Type *output_arg = static_cast<const our::Memory_Type *>(DMAHelper::base(&output));

our_cross_impl(in0_arg, in1_arg, output_arg);