parameters: (op)
return_statement: return _SquareMatrixSolveShapeHelper(
        op.inputs[0].get_shape().with_rank_at_least(2),
        op.inputs[1].get_shape().with_rank_at_least(2))
docstring: Shape function for batch square matrix solver ops.
docstring_summary: Shape function for batch square matrix solver ops.
docstring_tokens: Shape function for batch square matrix solver ops .
function: def _BatchSquareMatrixSolveShape(op):
    """Shape function for batch square matrix solver ops."""
    return _SquareMatrixSolveShapeHelper(
        op.inputs[0].get_shape().with_rank_at_least(2),
        op.inputs[1].get_shape().with_rank_at_least(2))
function_tokens: def _BatchSquareMatrixSolveShape ( op ) : return _SquareMatrixSolveShapeHelper ( op . inputs [ 0 ] . get_shape ( ) . with_rank_at_least ( 2 ) , op . inputs [ 1 ] . get_shape ( ) . with_rank_at_least ( 2 ) )