↓ 90 callersMethod_assert_slow_fast_tensors_equivalence(self, slow_tensor, fast_tensor, atol=1e-1, rtol=1e-3, mean_atol=5e-3)
Logics-Parsing-Omni/transformers-omni3/tests/test_image_processing_common.py:174
↓ 74 callersFunction_prepare_4d_attention_mask Creates a non-causal 4D mask of shape `(batch_size, 1, query_length, key_value_length)` from a 2D mask of shape `(batch_size, key_value_lengt
Logics-Parsing-Omni/transformers-omni3/src/transformers/modeling_attn_mask_utils.py:419