Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/ao/sparsity/
H A Dtest_qlinear_packed_params.py10 override_cpu_allocator_for_qnnpack,
187 with override_cpu_allocator_for_qnnpack(qengine_is_qnnpack()):
266 with override_cpu_allocator_for_qnnpack(qengine_is_qnnpack()):
276 with override_cpu_allocator_for_qnnpack(qengine_is_qnnpack()):
H A Dtest_kernels.py15 override_cpu_allocator_for_qnnpack,
74 with override_cpu_allocator_for_qnnpack(qengine_is_qnnpack()):
156 with override_cpu_allocator_for_qnnpack(qengine_is_qnnpack()):
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantized.py157 def override_cpu_allocator_for_qnnpack(qengine_is_qnnpack): function