parameters: (graph, node)
return_statement: return _unary_op_flops(graph, node, ops_per_element=5)
docstring: Compute flops for Softmax operation.
docstring_summary: Compute flops for Softmax operation.
docstring_tokens: Compute flops for Softmax operation .
function: def _softmax_flops(graph, node):
    """Compute flops for Softmax operation."""
    # Softmax implemetation:
    #
    # Approximate flops breakdown:
    #   2*n          -- compute shifted logits
    #   n            -- exp of shifted logits
    #   2*n          -- compute softmax from exp of shifted logits
    return _unary_op_flops(graph, node, ops_per_element=5)
function_tokens: def _softmax_flops ( graph , node ) : # Softmax implemetation: # # Approximate flops breakdown: #   2*n          -- compute shifted logits #   n            -- exp of shifted logits #   2*n          -- compute softmax from exp of shifted logits return _unary_op_flops ( graph , node , ops_per_element = 5 )