Searched refs:padding_head (Results 1 – 2 of 2) sorted by relevance
/frameworks/ml/nn/common/include/ |
D | OperationsUtils.h | 208 bool isTransposeConv, int32_t* padding_head, 213 int32_t* padding_head, int32_t* padding_tail) { in calculateExplicitPadding() argument 215 /*isTransposeConv=*/false, padding_head, padding_tail); in calculateExplicitPadding() 219 int32_t padding_implicit, int32_t* padding_head, in calculateExplicitPadding() argument 221 calculateExplicitPadding(in_size, stride, 1, filter_size, padding_implicit, padding_head, in calculateExplicitPadding() 227 int32_t* padding_head, int32_t* padding_tail) { in calculateExplicitPaddingTransposeConv() argument 229 padding_implicit, /*isTransposeConv=*/true, padding_head, in calculateExplicitPaddingTransposeConv()
|
/frameworks/ml/nn/common/ |
D | OperationsUtils.cpp | 291 bool isTransposeConv, int32_t* padding_head, in calculateExplicitPaddingImpl() argument 293 *padding_head = 0; in calculateExplicitPaddingImpl() 302 *padding_head = (tmp - in_size) / 2; in calculateExplicitPaddingImpl() 303 *padding_tail = (tmp - in_size) - *padding_head; in calculateExplicitPaddingImpl() 307 *padding_tail = (tmp - in_size) - *padding_head; in calculateExplicitPaddingImpl()
|