Searched refs:dev_a_dev_ptrs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 420 ScratchSpace<uint8> dev_a_dev_ptrs = in PotrfBatchedImpl() local
423 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in PotrfBatchedImpl()
424 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in PotrfBatchedImpl()
429 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in PotrfBatchedImpl()
766 ScratchSpace<uint8> dev_a_dev_ptrs = in GetrfBatchedImpl() local
769 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrfBatchedImpl()
770 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in GetrfBatchedImpl()
775 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in GetrfBatchedImpl()
801 ScratchSpace<uint8> dev_a_dev_ptrs = in GetrsBatchedImpl() local
807 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrsBatchedImpl()
[all …]
|