(attention_scores, attention_mask)
| 39 | |
| 40 | |
| 41 | def attention_mask_func(attention_scores, attention_mask): |
| 42 | attention_scores.masked_fill_(attention_mask, -10000.0) |
| 43 | |
| 44 | return attention_scores |
| 45 | |
| 46 | |
| 47 | def get_linear_layer(rows, columns, init_method): |
nothing calls this directly
no outgoing calls
no test coverage detected