Searched refs:lhs_batch_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 72 ShapeHandle lhs_batch_shape; in BandedTriangularSolveShapeFn() local
76 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in BandedTriangularSolveShapeFn()
79 c, lhs_batch_shape, rhs_batch_shape, true, &output_batch_shape)); in BandedTriangularSolveShapeFn()
105 ShapeHandle lhs_batch_shape; in MatrixSolveShapeFn() local
108 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in MatrixSolveShapeFn()
112 c->Merge(lhs_batch_shape, rhs_batch_shape, &lhs_batch_shape)); in MatrixSolveShapeFn()
124 TF_RETURN_IF_ERROR(c->Concatenate(lhs_batch_shape, c->Vector(n), &out)); in MatrixSolveShapeFn()
138 ShapeHandle lhs_batch_shape; in MatrixTriangularSolveShapeFn() local
142 TF_RETURN_IF_ERROR(c->Subshape(lhs, 0, -2, &lhs_batch_shape)); in MatrixTriangularSolveShapeFn()
145 c, lhs_batch_shape, rhs_batch_shape, true, &output_batch_shape)); in MatrixTriangularSolveShapeFn()
[all …]
|