Home
last modified time | relevance | path

Searched defs:linear_weight (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcpu_profiling_allocator_test.cpp11 at::Tensor linear_weight, in run_with_control_flow()
41 at::Tensor linear_weight = at::rand({32, 3136}); in TEST() local
99 at::Tensor linear_weight = at::rand({32, 3136}); in TEST() local
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_optimize_for_mobile_preserve_debug_info.py73 linear_weight, argument
164 linear_weight, argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_xnnpack_integration.py615 linear_weight, argument
651 linear_weight, argument