Searched refs:make_test_cases_dynamic_shape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | test_case_gen.py | 78 def make_test_cases_dynamic_shape(*args): function
357 make_test_cases_dynamic_shape(Scalar(5), torch.arange(5)),
386 + make_test_cases_dynamic_shape(
407 + make_test_cases_dynamic_shape(
427 + make_test_cases_dynamic_shape(
443 + make_test_cases_dynamic_shape(x, torch.Tensor.detach(x)),
473 + make_test_cases_dynamic_shape(
493 + make_test_cases_dynamic_shape(x, StringArg("tanh"), m(x)),
505 make_test_cases_dynamic_shape(x, 0, m(x)),
516 + make_test_cases_dynamic_shape(x, torch.log(x)),
[all …]
|