Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/test/
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
/external/pytorch/test/jit/
Dtest_optimize_for_mobile_preserve_debug_info.py73 linear_weight, argument
164 linear_weight, argument
/external/pytorch/test/
Dtest_xnnpack_integration.py615 linear_weight, argument
651 linear_weight, argument