Method allocate_scoped_tensor

tensorflow/core/kernels/cuda_solvers.cc:273–281

Allocates a temporary tensor. The CudaSolver object maintains a TensorReference to the underlying Tensor to prevent it from being deallocated prematurely.

// TensorReference to the underlying Tensor to prevent it from being deallocated
// prematurely.
Status CudaSolver::allocate_scoped_tensor(DataType type,
                                          const TensorShape& shape,
                                          Tensor* out_temp) {
  const Status status = context_->allocate_temp(type, shape, out_temp);
  if (status.ok()) {
    scratch_tensor_refs_.emplace_back(*out_temp);
  }
  return status;
}

Status CudaSolver::forward_input_or_allocate_scoped_tensor(
    gtl::ArraySlice<int> candidate_input_indices, DataType type,