Home
last modified time | relevance | path

Searched refs:dilation_height_factor (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DConv2D.cpp65 int32_t dilation_width_factor = 1, dilation_height_factor = 1; member
83 dilation_height_factor = context->getInputValue<int32_t>(9); in initialize()
99 dilation_height_factor = context->getInputValue<int32_t>(12); in initialize()
113 calculateExplicitPadding(input_height, stride_height, dilation_height_factor, in initialize()
124 NN_RET_CHECK_GT(dilation_height_factor, 0); in initialize()
176 int32_t dilation_width_factor, int32_t dilation_height_factor) { in needim2colData() argument
185 const bool need_dilated_im2col = dilation_width_factor != 1 || dilation_height_factor != 1; in needim2colData()
195 int32_t dilation_width_factor, int32_t dilation_height_factor, int32_t activation, in convNhwc() argument
209 dilation_width_factor, dilation_height_factor); in convNhwc()
214 dilation_width_factor, dilation_height_factor, paddingWidth, paddingHeight, in convNhwc()
[all …]
DDepthwiseConv2D.cpp49 int32_t dilation_width_factor = 1, dilation_height_factor = 1; member
69 dilation_height_factor = context->getInputValue<int32_t>(10); in initialize()
86 dilation_height_factor = context->getInputValue<int32_t>(13); in initialize()
100 calculateExplicitPadding(input_height, stride_height, dilation_height_factor, in initialize()
111 NN_RET_CHECK_GT(dilation_height_factor, 0); in initialize()
149 .dilation_height_factor = static_cast<int16>(dilationHeightFactor), in depthwiseConvNhwc()
220 .dilation_height_factor = static_cast<int16>(dilationHeightFactor), in depthwiseConvNhwc()
452 int32_t effectiveFilterHeight = (filterHeight - 1) * param.dilation_height_factor + 1; in prepare()
459 computeOutSize(height, filterHeight, param.stride_height, param.dilation_height_factor, in prepare()
490 param.dilation_width_factor, param.dilation_height_factor, in execute()
[all …]
/packages/modules/NeuralNetworks/driver/sample_hidl/
DSampleDriverFloatXNNPACK.cpp344 int32_t dilation_height_factor) { in CheckConvolutionParams() argument
355 if (dilation_height_factor <= 0) { in CheckConvolutionParams()
363 int32_t dilation_height_factor, int32_t depth_multiplier, uint32_t output_channels) { in CheckDepthwiseConvolutionParams() argument
381 if (dilation_height_factor <= 0) { in CheckDepthwiseConvolutionParams()
724 int32_t dilation_height_factor = 1; in VisitConv2DNode() local
741 dilation_height_factor = getScalarData<int32_t>(operands[ins[12]]); in VisitConv2DNode()
752 dilation_height_factor = getScalarData<int32_t>(operands[ins[9]]); in VisitConv2DNode()
756 stride_width, stride_height, dilation_width_factor, dilation_height_factor)); in VisitConv2DNode()
775 static_cast<uint32_t>(dilation_height_factor), in VisitConv2DNode()
824 int32_t dilation_height_factor = 1; in VisitDepthwiseConv2DNode() local
[all …]
/packages/modules/NeuralNetworks/common/include/
DOperationsExecutionUtils.h228 int32_t dilation_height_factor, Shape* output);