Searched refs:assert_close_tensors (Results 1 – 1 of 1) sorted by relevance
199 def assert_close_tensors(tensor_a, tensor_b): function237 assert_close_tensors(nn_mha_output, composite_mha_output)