Searched refs:depth_multiplier (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | DepthwiseConv2D.cpp | 50 int32_t depth_multiplier; member
62 depth_multiplier = context->getInputValue<int32_t>(6); in initialize()
79 depth_multiplier = context->getInputValue<int32_t>(9); in initialize()
112 NN_RET_CHECK_GT(depth_multiplier, 0); in initialize()
150 .depth_multiplier = static_cast<int16>(depthMultiplier), in depthwiseConvNhwc()
221 .depth_multiplier = static_cast<int16>(depthMultiplier), in depthwiseConvNhwc()
450 NN_OPS_CHECK(param.depth_multiplier * channels_in == channels_out); in prepare()
491 param.depth_multiplier, param.activation, param.useNchw, in execute()
504 param.depth_multiplier, param.activation, param.useNchw, in execute()
522 param.dilation_height_factor, param.depth_multiplier, param.activation, in execute()
[all …]
|
| /packages/modules/NeuralNetworks/driver/sample_hidl/ |
| D | SampleDriverFloatXNNPACK.cpp | 363 int32_t dilation_height_factor, int32_t depth_multiplier, uint32_t output_channels) { in CheckDepthwiseConvolutionParams() argument
371 if (depth_multiplier <= 0) { in CheckDepthwiseConvolutionParams()
374 if (output_channels % depth_multiplier != 0) { in CheckDepthwiseConvolutionParams()
822 int32_t stride_width, stride_height, depth_multiplier, activation; in VisitDepthwiseConv2DNode() local
838 depth_multiplier = getScalarData<int32_t>(operands[ins[9]]); in VisitDepthwiseConv2DNode()
850 depth_multiplier = getScalarData<int32_t>(operands[ins[6]]); in VisitDepthwiseConv2DNode()
868 depth_multiplier, output_channels)); in VisitDepthwiseConv2DNode()
878 static_cast<uint32_t>(depth_multiplier), in VisitDepthwiseConv2DNode()
880 static_cast<uint32_t>(output_channels / depth_multiplier), outputMin, outputMax, in VisitDepthwiseConv2DNode()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsExecutionUtils.h | 227 int32_t depth_multiplier, int32_t dilation_width_factor,
|