Searched refs:make_test_cases_broadcast_two_input_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | test_case_gen.py | 104 def make_test_cases_broadcast_two_input_tensor(x, y, cpp_args, torch_args, torch_fn): function
691 + make_test_cases_broadcast_two_input_tensor(x, y, (1,), (), torch_fn=torch.add)
719 make_test_cases_broadcast_two_input_tensor(x, y, (), (), torch_fn=torch.div)
741 make_test_cases_broadcast_two_input_tensor(
776 make_test_cases_broadcast_two_input_tensor(x, y, (), (), torch_fn=torch.mul)
803 make_test_cases_broadcast_two_input_tensor(x, y, (1,), (), torch_fn=torch.sub)
865 make_test_cases_broadcast_two_input_tensor(
|