Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dfbgemm_utils.cpp223 Tensor ConvertToChannelsLast3dTensor(const Tensor& src) { in ConvertToChannelsLast3dTensor() function
339 return ConvertToChannelsLast3dTensor(src); in ConvertConvWeightsToChannelLastTensor()
H A Dfbgemm_utils.h341 Tensor ConvertToChannelsLast3dTensor(const Tensor& src);
H A Dqconv.cpp290 : at::native::fbgemm_utils::ConvertToChannelsLast3dTensor(act); in apply_impl()