| 436 | #if GOOGLE_CUDA |
| 437 | template <typename Scalar> |
| 438 | ScratchSpace<Scalar> CudaSolver::GetScratchSpace(const TensorShape& shape, |
| 439 | const string& debug_info, |
| 440 | bool on_host) { |
| 441 | ScratchSpace<Scalar> new_scratch_space(context_, shape, debug_info, on_host); |
| 442 | scratch_tensor_refs_.emplace_back(new_scratch_space.tensor()); |
| 443 | return std::move(new_scratch_space); |
| 444 | } |
| 445 | |
| 446 | template <typename Scalar> |
| 447 | ScratchSpace<Scalar> CudaSolver::GetScratchSpace(int64 size, |
nothing calls this directly
no test coverage detected