Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dtridiagonal_matmul_op.cc36 void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function in tensorflow::TridiagonalMatMulOp
H A Dsvd_op_impl.h50 void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function
H A Dmatrix_solve_ls_op_impl.h53 void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function
H A Dcholesky_grad.cc41 void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function in tensorflow::CholeskyGrad
H A Dlinalg_ops_common.h57 virtual void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function
H A Dlinalg_ops_common.cc164 ValidateInputMatrixShapes(context, *input_matrix_shapes); in AnalyzeInputs()
H A Dqr_op_impl.h64 void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function
H A Dmatrix_solve_op.cc53 void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function in tensorflow::MatrixSolveOp
H A Dtridiagonal_solve_op_gpu.cu.cc111 void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function in tensorflow::TridiagonalSolveOpGpuLinalg
H A Dtridiagonal_solve_op.cc58 void ValidateInputMatrixShapes( in ValidateInputMatrixShapes() function in tensorflow::TridiagonalSolveOp