Searched defs:rhs_batch_shape (Results 1 – 1 of 1) sorted by relevance
73 ShapeHandle rhs_batch_shape; in BandedTriangularSolveShapeFn() local106 ShapeHandle rhs_batch_shape; in MatrixSolveShapeFn() local139 ShapeHandle rhs_batch_shape; in MatrixTriangularSolveShapeFn() local300 ShapeHandle rhs_batch_shape; in TridiagonalMatMulShapeFn() local341 ShapeHandle rhs_batch_shape; in TridiagonalSolveShapeFn() local