Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dfbgemm_utils.cpp259 auto fused_tensor = at::cat(oc_g_ic_g_hw_tensors, 1); in TransposeConvTensorUnpackConversion() local
306 auto fused_tensor = at::cat(oc_g_ic_g_hw_tensors, 1); in TransposeConvTensorUnpackConversion() local
324 auto fused_tensor = at::cat(ic_g_oc_g_hw_tensors); in ConvertConvWeightsToChannelLastTensor() local
H A DXnnpackUtils.cpp77 auto fused_tensor = at::cat(ic_g_oc_g_hw_tensors); in convert_conv_weights_to_channel_last_tensor() local