Searched defs:wts (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleKernel.cpp | 80 opmath_t wts = *(scalar_t*)&data[1][i * strides[1]]; in eval() local
97 opmath_t wts = *(scalar_t*)&data[1][i * strides[1]]; in eval() local
178 scalar_t wts = wts_ptr[0]; in interpolate_aa_single_dim_zero_strides() local
204 scalar_t wts = wts_ptr[0]; in interpolate_aa_single_dim() local
342 int16_t wts = wts_ptr[0]; in basic_loop_aa_vertical() local
409 int16_t wts = wts_ptr[0]; in basic_loop_aa_horizontal() local
889 auto wts = empty(new_shape, CPU(c10::CppTypeToScalarType<scalar_t>())); in _compute_index_ranges_weights() local
|
| /aosp_15_r20/external/strace/ |
| H A D | count.c | 67 struct timespec wts; in count_syscall() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSample.cuh | 355 scalar_t wts = static_cast<accscalar_t>(weights[0]); in interpolate_aa_single_dim() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 2410 def helper(shape, eps=1, momentum=0.1, wts=False, training=False, channels_last=False, argument
2742 …def helper(shape, eps=1, momentum=0.1, wts=False, channels_last=False, track_running_stats=True, t… argument
|