Home
last modified time | relevance | path

Searched refs:padding_head (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/include/
DOperationsUtils.h212 bool isTransposeConv, int32_t* padding_head,
217 int32_t* padding_head, int32_t* padding_tail) { in calculateExplicitPadding() argument
219 /*isTransposeConv=*/false, padding_head, padding_tail); in calculateExplicitPadding()
223 int32_t padding_implicit, int32_t* padding_head, in calculateExplicitPadding() argument
225 calculateExplicitPadding(in_size, stride, 1, filter_size, padding_implicit, padding_head, in calculateExplicitPadding()
231 int32_t* padding_head, int32_t* padding_tail) { in calculateExplicitPaddingTransposeConv() argument
233 padding_implicit, /*isTransposeConv=*/true, padding_head, in calculateExplicitPaddingTransposeConv()
/frameworks/ml/nn/common/
DOperationsUtils.cpp320 bool isTransposeConv, int32_t* padding_head, in calculateExplicitPaddingImpl() argument
322 *padding_head = 0; in calculateExplicitPaddingImpl()
331 *padding_head = (tmp - in_size) / 2; in calculateExplicitPaddingImpl()
332 *padding_tail = (tmp - in_size) - *padding_head; in calculateExplicitPaddingImpl()
336 *padding_tail = (tmp - in_size) - *padding_head; in calculateExplicitPaddingImpl()