Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc850 ScratchSpace<uint8> dev_a_inv_dev_ptrs = cuda_solver->GetScratchSpace<uint8>( in GetriBatchedImpl() local
854 !CopyHostToDevice(context, dev_a_inv_dev_ptrs.mutable_data(), in GetriBatchedImpl()
855 host_a_inv_dev_ptrs, dev_a_inv_dev_ptrs.bytes())) { in GetriBatchedImpl()
862 reinterpret_cast<CudaScalar**>(dev_a_inv_dev_ptrs.mutable_data()), in GetriBatchedImpl()
893 ScratchSpace<uint8> dev_a_inv_dev_ptrs = cuda_solver->GetScratchSpace<uint8>( in MatInvBatchedImpl() local
897 !CopyHostToDevice(context, dev_a_inv_dev_ptrs.mutable_data(), in MatInvBatchedImpl()
898 host_a_inv_dev_ptrs, dev_a_inv_dev_ptrs.bytes())) { in MatInvBatchedImpl()
904 reinterpret_cast<CudaScalar**>(dev_a_inv_dev_ptrs.mutable_data()), ldainv, in MatInvBatchedImpl()