parameters: (op)
return_statement: return [tensor_shape.vector(batch_size.value)]
docstring: Shape function for InTopK op.
docstring_summary: Shape function for InTopK op.
docstring_tokens: Shape function for InTopK op .
function: def _InTopKShape(op):
    """Shape function for InTopK op."""
    predictions_shape = op.inputs[0].get_shape().with_rank(2)
    targets_shape = op.inputs[1].get_shape().with_rank(1)
    batch_size = predictions_shape[0].merge_with(targets_shape[0])
    return [tensor_shape.vector(batch_size.value)]
function_tokens: def _InTopKShape ( op ) : predictions_shape = op . inputs [ 0 ] . get_shape ( ) . with_rank ( 2 ) targets_shape = op . inputs [ 1 ] . get_shape ( ) . with_rank ( 1 ) batch_size = predictions_shape [ 0 ] . merge_with ( targets_shape [ 0 ] ) return [ tensor_shape . vector ( batch_size . value ) ]