Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_solve_op_gpu.cu.cc227 bool* not_invertible_dev; in SolveForSizeOneOrTwo() local
228 cudaMalloc(&not_invertible_dev, sizeof(bool)); in SolveForSizeOneOrTwo()
232 not_invertible_dev)); in SolveForSizeOneOrTwo()
234 cudaMemcpy(&not_invertible_host, not_invertible_dev, sizeof(bool), in SolveForSizeOneOrTwo()
236 cudaFree(not_invertible_dev); in SolveForSizeOneOrTwo()