parameters: (op)
return_statement: return [tensor_shape.matrix(None, input_shape.ndims)]
docstring: Shape function for the Where op.
docstring_summary: Shape function for the Where op.
docstring_tokens: Shape function for the Where op .
function: def _WhereShape(op):
    """Shape function for the Where op."""
    input_shape = op.inputs[0].get_shape()
    return [tensor_shape.matrix(None, input_shape.ndims)]
function_tokens: def _WhereShape ( op ) : input_shape = op . inputs [ 0 ] . get_shape ( ) return [ tensor_shape . matrix ( None , input_shape . ndims ) ]