OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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