Searched refs:rhs_batch_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 73 ShapeHandle rhs_batch_shape; in BandedTriangularSolveShapeFn() local
77 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in BandedTriangularSolveShapeFn()
79 c, lhs_batch_shape, rhs_batch_shape, true, &output_batch_shape)); in BandedTriangularSolveShapeFn()
106 ShapeHandle rhs_batch_shape; in MatrixSolveShapeFn() local
109 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in MatrixSolveShapeFn()
112 c->Merge(lhs_batch_shape, rhs_batch_shape, &lhs_batch_shape)); in MatrixSolveShapeFn()
139 ShapeHandle rhs_batch_shape; in MatrixTriangularSolveShapeFn() local
143 TF_RETURN_IF_ERROR(c->Subshape(rhs, 0, -2, &rhs_batch_shape)); in MatrixTriangularSolveShapeFn()
145 c, lhs_batch_shape, rhs_batch_shape, true, &output_batch_shape)); in MatrixTriangularSolveShapeFn()
300 ShapeHandle rhs_batch_shape; in TridiagonalMatMulShapeFn() local
[all …]
|