Home
last modified time | relevance | path

Searched refs:stride_height (Results 1 – 11 of 11) sorted by relevance

/packages/modules/NeuralNetworks/common/operations/
DConv2D.cpp66 int32_t stride_width, stride_height; member
78 stride_height = context->getInputValue<int32_t>(5); in initialize()
94 stride_height = context->getInputValue<int32_t>(8); in initialize()
115 calculateExplicitPadding(input_height, stride_height, dilation_height_factor, in initialize()
124 NN_RET_CHECK_GT(stride_height, 0); in initialize()
178 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData() argument
190 stride_width != 1 || stride_height != 1 || filter_width != 1 || filter_height != 1; in needim2colData()
197 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in convNhwc() argument
211 const bool need_im2colData = needim2colData(filterShape, stride_width, stride_height, in convNhwc()
216 biasData, convertShapeToDims(biasShape), stride_width, stride_height, in convNhwc()
[all …]
DGroupedConv2D.cpp49 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvFloat32() argument
69 static_cast<int32_t>(h) * stride_height - padding_top; in groupedConvFloat32()
107 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvQuant8() argument
141 static_cast<int32_t>(h) * stride_height - padding_top; in groupedConvQuant8()
186 int32_t stride_width, int32_t stride_height,
195 int32_t stride_width, int32_t stride_height,
205 int32_t stride_width, int32_t stride_height, int32_t numGroups, in groupedConvQuant8PerChannel() argument
245 static_cast<int32_t>(h) * stride_height - padding_top; in groupedConvQuant8PerChannel()
289 int32_t stride_width, int32_t stride_height, int32_t numGroups, in groupedConvFloat16() argument
304 padding_bottom, stride_width, stride_height, numGroups, activation, in groupedConvFloat16()
[all …]
DDepthwiseConv2D.cpp52 int32_t stride_width, stride_height; member
65 stride_height = context->getInputValue<int32_t>(5); in initialize()
82 stride_height = context->getInputValue<int32_t>(8); in initialize()
104 calculateExplicitPadding(input_height, stride_height, dilation_height_factor, in initialize()
113 NN_RET_CHECK_GT(stride_height, 0); in initialize()
151 .stride_height = static_cast<int16>(strideHeight), in depthwiseConvNhwc()
222 .stride_height = static_cast<int16>(strideHeight), in depthwiseConvNhwc()
562 computeOutSize(height, filterHeight, param.stride_height, param.dilation_height_factor, in prepare()
592 param.stride_width, param.stride_height, in execute()
605 param.stride_width, param.stride_height, in execute()
[all …]
DPooling.cpp48 int32_t stride_width, stride_height; member
62 stride_height = context->getInputValue<int32_t>(6); in initialize()
72 stride_height = context->getInputValue<int32_t>(3); in initialize()
86 calculateExplicitPadding(input_height, stride_height, filter_height, padding_implicit, in initialize()
94 NN_RET_CHECK_GT(stride_height, 0); in initialize()
111 .stride_height = stride_height, in toTfliteParam()
379 uint32_t outHeight = computeOutSize(height, param.filter_height, param.stride_height, in prepare()
/packages/modules/NeuralNetworks/driver/sample/
DSampleDriverFloatXNNPACK.cpp342 static V1_3::ErrorStatus CheckConvolutionParams(int32_t stride_width, int32_t stride_height, in CheckConvolutionParams() argument
348 if (stride_height <= 0) { in CheckConvolutionParams()
362 int32_t stride_width, int32_t stride_height, int32_t dilation_width_factor, in CheckDepthwiseConvolutionParams() argument
367 if (stride_height <= 0) { in CheckDepthwiseConvolutionParams()
388 static V1_3::ErrorStatus CheckPoolingParams(int32_t stride_width, int32_t stride_height, in CheckPoolingParams() argument
393 if (stride_height <= 0) { in CheckPoolingParams()
403 if (filter_width == 1 && filter_height == 1 && std::max(stride_width, stride_height) > 1) { in CheckPoolingParams()
635 int32_t stride_width, stride_height, filter_width, filter_height, activation; in VisitAveragePool2DNode() local
648 stride_height = getScalarData<int32_t>(operands[ins[6]]); in VisitAveragePool2DNode()
657 stride_height = getScalarData<int32_t>(operands[ins[3]]); in VisitAveragePool2DNode()
[all …]
/packages/modules/NeuralNetworks/common/include/
DOperations.h152 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
159 int32_t stride_height, int32_t activation, float* outputData,
167 int32_t stride_height, int32_t activation, T* outputData,
176 int32_t stride_width, int32_t stride_height, int32_t numGroups,
DOperationsUtils.h284 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
321 int32_t padding_bottom, int32_t stride_width, int32_t stride_height,
/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
Dl2_pool_float_large.mod.py22 stride_height = Int32Scalar("stride_height", 1) variable
27 stride_width, stride_height,
/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
Dl2_pool_float_large_relaxed.mod.py22 stride_height = Int32Scalar("stride_height", 1) variable
27 stride_width, stride_height,
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1471 int32_t stride_width, stride_height; in executeOperation() local
1482 stride_height = getScalarData<int32_t>(operands[ins[8]]); in executeOperation()
1489 stride_height = getScalarData<int32_t>(operands[ins[5]]); in executeOperation()
1517 calculateExplicitPadding(input_height, stride_height, filter_height, in executeOperation()
1523 stride_height, numGroups, &outShape) || in executeOperation()
1535 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1543 padding_right, padding_top, padding_bottom, stride_width, stride_height, in executeOperation()
1555 stride_height, numGroups, activation, in executeOperation()
1563 stride_height, numGroups, activation, in executeOperation()
1575 stride_height, numGroups, activation, in executeOperation()
[all …]
DOperationsUtils.cpp709 int32_t padding_bottom, int32_t stride_width, int32_t stride_height, in groupedConvPrepare() argument
747 computeOutSize(height, filterHeight, stride_height, padding_top, padding_bottom); in groupedConvPrepare()