Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_matmul_op.cc126 REGISTER_LINALG_OP_CPU("TridiagonalMatMul", (TridiagonalMatMulOp<float>),
128 REGISTER_LINALG_OP_CPU("TridiagonalMatMul", (TridiagonalMatMulOp<double>),
130 REGISTER_LINALG_OP_CPU("TridiagonalMatMul", (TridiagonalMatMulOp<complex64>),
132 REGISTER_LINALG_OP_CPU("TridiagonalMatMul", (TridiagonalMatMulOp<complex128>),
Dtridiagonal_solve_op.cc226 REGISTER_LINALG_OP_CPU("TridiagonalSolve", (TridiagonalSolveOp<float>), float);
227 REGISTER_LINALG_OP_CPU("TridiagonalSolve", (TridiagonalSolveOp<double>),
229 REGISTER_LINALG_OP_CPU("TridiagonalSolve", (TridiagonalSolveOp<complex64>),
231 REGISTER_LINALG_OP_CPU("TridiagonalSolve", (TridiagonalSolveOp<complex128>),
Dlinalg_ops_common.h209 #define REGISTER_LINALG_OP_CPU(OpName, OpClass, Scalar) \ macro
219 REGISTER_LINALG_OP_CPU(OpName, OpClass, Scalar)