Function TF_AllocateOutput

tensorflow/c/kernels.cc:245–254

}

TF_Tensor* TF_AllocateOutput(TF_OpKernelContext* context, int index,
                             TF_DataType dtype, int64_t* dims, int num_dims,
                             size_t len) {
  auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(context);
  tensorflow::AllocatorAttributes attr = cc_ctx->output_alloc_attr(index);
  auto* allocator = cc_ctx->get_allocator(attr);
  void* data = tensorflow::allocate_tensor("TF_AllocateOutput", len, allocator);
  return TF_NewTensor(dtype, dims, num_dims, data, len,
                      tensorflow::deallocate_buffer, allocator);
}