Searched refs:dev_b_dev_ptrs (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | cuda_solvers.cc | 811 ScratchSpace<uint8> dev_b_dev_ptrs = in GetrsBatchedImpl() local
818 if (!CopyHostToDevice(context, dev_b_dev_ptrs.mutable_data() /* dest */, in GetrsBatchedImpl()
819 host_b_dev_ptrs /* source */, dev_b_dev_ptrs.bytes())) { in GetrsBatchedImpl()
825 dev_pivots, reinterpret_cast<CudaScalar**>(dev_b_dev_ptrs.mutable_data()), in GetrsBatchedImpl()
996 ScratchSpace<uint8> dev_b_dev_ptrs = in TrsmBatchedImpl() local
1003 if (!CopyHostToDevice(context, dev_b_dev_ptrs.mutable_data() /* dest */, in TrsmBatchedImpl()
1004 host_b_dev_ptrs /* source */, dev_b_dev_ptrs.bytes())) { in TrsmBatchedImpl()
1011 lda, reinterpret_cast<CudaScalar**>(dev_b_dev_ptrs.mutable_data()), in TrsmBatchedImpl()
|