Searched refs:dev_a_dev_ptrs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 708 ScratchSpace<uint8> dev_a_dev_ptrs = in GetrfBatchedImpl() local
711 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrfBatchedImpl()
712 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in GetrfBatchedImpl()
717 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in GetrfBatchedImpl()
743 ScratchSpace<uint8> dev_a_dev_ptrs = in GetrsBatchedImpl() local
749 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrsBatchedImpl()
750 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in GetrsBatchedImpl()
759 reinterpret_cast<const CudaScalar* const*>(dev_a_dev_ptrs.data()), lda, in GetrsBatchedImpl()
789 ScratchSpace<uint8> dev_a_dev_ptrs = in GetriBatchedImpl() local
794 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetriBatchedImpl()
[all …]
|