HomeSort by relevance Sort by last modified time
    Searched refs:DeviceLapackInfo (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
cuda_solvers.h 70 class DeviceLapackInfo;
106 // std::vector<DeviceLapackInfo> dev_info;
148 const std::vector<DeviceLapackInfo>& dev_lapack_info,
157 const std::vector<DeviceLapackInfo>& dev_lapack_info,
168 // Returns a DeviceLapackInfo that will live for the duration of the
170 inline DeviceLapackInfo GetDeviceLapackInfo(int64 size,
231 int* dev_pivots, DeviceLapackInfo* dev_lapack_info,
241 int ldb, DeviceLapackInfo* dev_lapack_info,
252 DeviceLapackInfo* dev_lapack_info,
261 int ldainv, DeviceLapackInfo* dev_lapack_info
    [all...]
cuda_solvers.cc 181 const std::vector<DeviceLapackInfo>& dev_lapack_infos,
253 const std::vector<DeviceLapackInfo>& dev_lapack_info,
660 DeviceLapackInfo* dev_lapack_info,
    [all...]
svd_op_gpu.cu.cc 172 const std::vector<DeviceLapackInfo>& dev_info,
245 std::vector<DeviceLapackInfo> dev_info;
318 std::vector<DeviceLapackInfo> dev_info;
cholesky_op.cc 145 std::vector<DeviceLapackInfo> dev_info;
self_adjoint_eig_v2_op_gpu.cc 125 std::vector<DeviceLapackInfo> dev_info;
determinant_op.cc 203 std::vector<DeviceLapackInfo> dev_info;
347 std::vector<DeviceLapackInfo> dev_info;
matrix_inverse_op.cc 165 std::vector<DeviceLapackInfo> dev_info;
matrix_solve_op.cc 213 std::vector<DeviceLapackInfo> dev_info;
qr_op_impl.h 199 std::vector<DeviceLapackInfo> dev_info;

Completed in 373 milliseconds