Searched refs:padding_head (Results 1 – 2 of 2) sorted by relevance
/frameworks/ml/nn/common/include/ |
D | OperationsUtils.h | 212 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/ |
D | OperationsUtils.cpp | 320 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()
|