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

  /external/tensorflow/tensorflow/core/kernels/
qr_op_double.cc 23 REGISTER_LINALG_OP_GPU("Qr", (QrOpGpu<double>), double);
qr_op_float.cc 23 REGISTER_LINALG_OP_GPU("Qr", (QrOpGpu<float>), float);
qr_op_complex128.cc 23 REGISTER_LINALG_OP_GPU("Qr", (QrOpGpu<complex128>), complex128);
qr_op_complex64.cc 23 REGISTER_LINALG_OP_GPU("Qr", (QrOpGpu<complex64>), complex64);
cholesky_op.cc 170 REGISTER_LINALG_OP_GPU("Cholesky", (CholeskyOpGpu<float>), float);
171 REGISTER_LINALG_OP_GPU("Cholesky", (CholeskyOpGpu<double>), double);
172 REGISTER_LINALG_OP_GPU("Cholesky", (CholeskyOpGpu<complex64>), complex64);
173 REGISTER_LINALG_OP_GPU("Cholesky", (CholeskyOpGpu<complex128>), complex128);
matrix_triangular_solve_op.cc 246 REGISTER_LINALG_OP_GPU("MatrixTriangularSolve",
248 REGISTER_LINALG_OP_GPU("MatrixTriangularSolve",
250 REGISTER_LINALG_OP_GPU("MatrixTriangularSolve",
252 REGISTER_LINALG_OP_GPU("MatrixTriangularSolve",
254 REGISTER_LINALG_OP_GPU("BatchMatrixTriangularSolve",
256 REGISTER_LINALG_OP_GPU("BatchMatrixTriangularSolve",
determinant_op.cc 414 REGISTER_LINALG_OP_GPU("MatrixDeterminant", (DeterminantOpGpu<float>), float);
415 REGISTER_LINALG_OP_GPU("MatrixDeterminant", (DeterminantOpGpu<double>), double);
416 REGISTER_LINALG_OP_GPU("MatrixDeterminant", (DeterminantOpGpu<complex64>),
418 REGISTER_LINALG_OP_GPU("MatrixDeterminant", (DeterminantOpGpu<complex128>),
421 REGISTER_LINALG_OP_GPU("LogMatrixDeterminant", (LogDeterminantOpGpu<float>),
423 REGISTER_LINALG_OP_GPU("LogMatrixDeterminant", (LogDeterminantOpGpu<double>),
425 REGISTER_LINALG_OP_GPU("LogMatrixDeterminant", (LogDeterminantOpGpu<complex64>),
427 REGISTER_LINALG_OP_GPU("LogMatrixDeterminant",
matrix_inverse_op.cc 269 REGISTER_LINALG_OP_GPU("MatrixInverse", (MatrixInverseOpGpu<float>), float);
270 REGISTER_LINALG_OP_GPU("MatrixInverse", (MatrixInverseOpGpu<double>), double);
271 REGISTER_LINALG_OP_GPU("MatrixInverse", (MatrixInverseOpGpu<complex64>),
273 REGISTER_LINALG_OP_GPU("MatrixInverse", (MatrixInverseOpGpu<complex128>),
matrix_solve_op.cc 354 REGISTER_LINALG_OP_GPU("MatrixSolve", (MatrixSolveOpGpu<float>), float);
355 REGISTER_LINALG_OP_GPU("MatrixSolve", (MatrixSolveOpGpu<double>), double);
356 REGISTER_LINALG_OP_GPU("MatrixSolve", (MatrixSolveOpGpu<complex64>), complex64);
357 REGISTER_LINALG_OP_GPU("MatrixSolve", (MatrixSolveOpGpu<complex128>),
svd_op_gpu.cu.cc 408 REGISTER_LINALG_OP_GPU("Svd", (SvdOpGpu<float>), float);
409 REGISTER_LINALG_OP_GPU("Svd", (SvdOpGpu<double>), double);
412 REGISTER_LINALG_OP_GPU("BatchSvd", (SvdOpGpu<float>), float);
413 REGISTER_LINALG_OP_GPU("BatchSvd", (SvdOpGpu<double>), double);
linalg_ops_common.h 189 #define REGISTER_LINALG_OP_GPU(OpName, OpClass, Scalar) \

Completed in 240 milliseconds