Searched refs:depth_multiplier (Results 1 – 4 of 4) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | DepthwiseConv2D.cpp | 43 int32_t depth_multiplier, int32_t activation, in depthwiseConvFloat32() argument 57 paddingWidth, paddingHeight, depth_multiplier, in depthwiseConvFloat32() 71 int32_t depth_multiplier, int32_t activation, in depthwiseConvQuant8() argument 102 paddingWidth, paddingHeight, depth_multiplier, in depthwiseConvQuant8()
|
/frameworks/ml/nn/common/include/ |
D | Operations.h | 69 int32_t depth_multiplier, int32_t activation, 77 int32_t depth_multiplier, int32_t activation,
|
/frameworks/ml/nn/common/ |
D | CpuExecutor.cpp | 471 int32_t depth_multiplier; in executeOperation() local 481 depth_multiplier = getScalarData<int32_t>(mOperands[ins[9]]); in executeOperation() 487 depth_multiplier = getScalarData<int32_t>(mOperands[ins[6]]); in executeOperation() 523 depth_multiplier, activation, in executeOperation() 542 depth_multiplier, activation, in executeOperation()
|
/frameworks/ml/nn/tools/test_generator/ |
D | test_generator.py | 489 …onv(self, filter, bias, input, padding, stride_width, stride_height, depth_multiplier, activation): argument 491 stride_height, depth_multiplier, activation]
|