Home
last modified time | relevance | path

Searched refs:generate_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/lazy/
H A Dtest_generator.py20 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 Dtest_gemm.py430 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 Dtest_linalg.py6920 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():