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():
|