Searched refs:batch_ptr_copy (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 253 GpuDeviceArrayOnHost<int> batch_ptr_copy(c, host_batch_ptr.size()); in operator ()() local
254 TF_RETURN_IF_ERROR(batch_ptr_copy.Init()); in operator ()()
256 batch_ptr_copy.Set(i, host_batch_ptr(i)); in operator ()()
258 TF_RETURN_IF_ERROR(batch_ptr_copy.Finalize()); in operator ()()
266 batch_ptr_copy.data(), batch_size, size)); in operator ()()
321 GpuDeviceArrayOnHost<int> batch_ptr_copy(ctx, host_batch_ptr.size()); in CSRSparseMatrixBatchMulVecImpl() local
322 TF_RETURN_IF_ERROR(batch_ptr_copy.Init()); in CSRSparseMatrixBatchMulVecImpl()
324 batch_ptr_copy.Set(i, host_batch_ptr(i)); in CSRSparseMatrixBatchMulVecImpl()
326 TF_RETURN_IF_ERROR(batch_ptr_copy.Finalize()); in CSRSparseMatrixBatchMulVecImpl()
335 b.data(), c_values.data(), batch_ptr_copy.data(), batch_size, total_nnz)); in CSRSparseMatrixBatchMulVecImpl()
[all …]
|