Home
last modified time | relevance | path

Searched defs:max_bvec (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveMaxPoolKernel.cpp319 bVec max_bvec = convert_from_float<scalar_t>(max_fvec0, max_fvec1); in cpu_adaptive_max_pool2d_channels_last() local
809 bVec max_bvec = convert_float_bfloat16(max_fvec0, max_fvec1); in cpu_adaptive_max_pool3d_channels_last() local
H A DMaxPoolKernel.cpp227 Vec max_bvec = convert_from_float<scalar_t>(max_fvec0, max_fvec1); in compute_internal() local