Home
last modified time | relevance | path

Searched refs:dev_b_dev_ptrs (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc804 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()