Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/xnnpack/
DAveragePooling.cpp48 size_t workspace_alignment = 0; in global_average_pool() local
61 &workspace_alignment, // workspace_alignment in global_average_pool()
70 std::vector<char> workspace_vector(workspace_size + workspace_alignment + xnnpack_buffer_padding); in global_average_pool()
73 …)((intptr_t)maybe_aligned_workspace + workspace_alignment - (intptr_t)maybe_aligned_workspace % wo… in global_average_pool()
DConvolution.cpp360 size_t workspace_alignment = SIZE_MAX; in run() local
368 &workspace_alignment, // workspace_alignment in run()
/external/pytorch/aten/src/ATen/native/quantized/cpu/
DXnnpackUtils.h200 size_t workspace_alignment = SIZE_MAX; variable
209 &workspace_alignment, /* size_t* workspace_alignment */
220 &workspace_alignment, /* size_t* workspace_alignment */