Searched refs:generate_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/lazy/ |
| H A D | test_generator.py | 20 def generate_tensor(): function
36 cpu_t1, cpu_t2 = generate_tensor()
41 lazy_t1, lazy_t2 = generate_tensor()
58 def generate_tensor(seed): function
68 t = generate_tensor(1)
76 t = generate_tensor(2)
84 t = generate_tensor(1)
|
| /aosp_15_r20/external/pytorch/test/xpu/ |
| H A D | test_gemm.py | 430 def generate_tensor(): function
520 for b1, b2, ref, out_tensor in generate_tensor():
533 def generate_tensor(): function
584 for b1, b2, ref, out_tensor in generate_tensor():
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_linalg.py | 6920 def generate_tensor(): function
6957 for b1, b2, ref, out_tensor in generate_tensor():
6991 def generate_tensor(): function
7025 for b1, b2, ref, out_tensor in generate_tensor():
|