Searched refs:dev_a_dev_ptrs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 418 ScratchSpace<uint8> dev_a_dev_ptrs = in PotrfBatchedImpl() local
421 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in PotrfBatchedImpl()
422 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in PotrfBatchedImpl()
427 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in PotrfBatchedImpl()
773 ScratchSpace<uint8> dev_a_dev_ptrs = in GetrfBatchedImpl() local
776 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrfBatchedImpl()
777 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in GetrfBatchedImpl()
782 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in GetrfBatchedImpl()
808 ScratchSpace<uint8> dev_a_dev_ptrs = in GetrsBatchedImpl() local
814 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrsBatchedImpl()
[all …]
|