Searched refs:padding_head (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsUtils.h | 196 bool isTransposeConv, int32_t* padding_head,
201 int32_t* padding_head, int32_t* padding_tail) { in calculateExplicitPadding() argument
203 /*isTransposeConv=*/false, padding_head, padding_tail); in calculateExplicitPadding()
207 int32_t padding_implicit, int32_t* padding_head, in calculateExplicitPadding() argument
209 calculateExplicitPadding(in_size, stride, 1, filter_size, padding_implicit, padding_head, in calculateExplicitPadding()
215 int32_t* padding_head, int32_t* padding_tail) { in calculateExplicitPaddingTransposeConv() argument
217 padding_implicit, /*isTransposeConv=*/true, padding_head, in calculateExplicitPaddingTransposeConv()
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 318 bool isTransposeConv, int32_t* padding_head, in calculateExplicitPaddingImpl() argument
320 *padding_head = 0; in calculateExplicitPaddingImpl()
329 *padding_head = (tmp - in_size) / 2; in calculateExplicitPaddingImpl()
330 *padding_tail = (tmp - in_size) - *padding_head; in calculateExplicitPaddingImpl()
334 *padding_tail = (tmp - in_size) - *padding_head; in calculateExplicitPaddingImpl()
|