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,
|