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