Searched refs:MakeBatchSquareMatrix (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 29 Status MakeBatchSquareMatrix(InferenceContext* c, ShapeHandle input, in MakeBatchSquareMatrix() function
45 TF_RETURN_IF_ERROR(MakeBatchSquareMatrix(c, c->input(0), &out)); in BatchUnchangedSquareShapeFn()
99 TF_RETURN_IF_ERROR(MakeBatchSquareMatrix(c, c->input(0), &lhs)); in MatrixSolveShapeFn()
135 TF_RETURN_IF_ERROR(MakeBatchSquareMatrix(c, c->input(0), &lhs)); in MatrixTriangularSolveShapeFn()
163 TF_RETURN_IF_ERROR(MakeBatchSquareMatrix(c, c->input(0), &input)); in SelfAdjointEigV2ShapeFn()
444 TF_RETURN_IF_ERROR(MakeBatchSquareMatrix(c, c->input(0), &input)); in __anon8ebee8ea0402()
|