Home
last modified time | relevance | path

Searched refs:channels_in (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DTransposeConv2D.cpp463 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare() local
468 NN_RET_CHECK_EQ(channels_in, getSizeOfDimension(filter, 3)); in prepare()
472 NN_RET_CHECK_GT(channels_in, 0u); in prepare()
DConv2D.cpp557 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare() local
562 NN_RET_CHECK_EQ(channels_in, getSizeOfDimension(filter, 3)); in prepare()
566 NN_RET_CHECK_GT(channels_in, 0u); in prepare()
DDepthwiseConv2D.cpp445 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare() local
450 NN_OPS_CHECK(param.depth_multiplier * channels_in == channels_out); in prepare()