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