Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DDepthwiseConvKernel.cpp198 float32x4x4_t input_tile; in convolution_depthwise3x3_winograd_impl() local
224 float32x4x4_t input_tile; in convolution_depthwise3x3_winograd_impl() local
392 vfloat32m1x4_t input_tile; in convolution_depthwise3x3_winograd_impl() local
421 vfloat32m1x4_t input_tile; in convolution_depthwise3x3_winograd_impl() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DWinograd.cpp382 SimpleTensor<T> input_tile{ TensorShape(kernel_max_dim, kernel_max_dim), in.data_type(), 1 }; in winograd_filter_transform() local
475 SimpleTensor<T> input_tile{ in_tile_dims, in.data_type(), 1 }; in winograd_output_transform() local