Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dfbgemm_utils.cpp312 Tensor ConvertConvWeightsToChannelLastTensor<2>( in ConvertConvWeightsToChannelLastTensor() function
334 Tensor ConvertConvWeightsToChannelLastTensor<3>( in ConvertConvWeightsToChannelLastTensor() function
H A Dfbgemm_utils.h347 Tensor ConvertConvWeightsToChannelLastTensor(
H A Dqconv_prepack.cpp118 …at::native::fbgemm_utils::ConvertConvWeightsToChannelLastTensor<kSpatialDim>(weight, groups, trans… in prepack()