Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DTransposeConv2D.cpp510 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare() local
515 NN_RET_CHECK_EQ(channels_in, getSizeOfDimension(filter, 3)); in prepare()
519 NN_RET_CHECK_GT(channels_in, 0); in prepare()
DConv2D.cpp656 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare() local
661 NN_RET_CHECK_EQ(channels_in, getSizeOfDimension(filter, 3)); in prepare()
665 NN_RET_CHECK_GT(channels_in, 0); in prepare()
DDepthwiseConv2D.cpp540 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare() local
545 NN_OPS_CHECK(param.depth_multiplier * channels_in == channels_out); in prepare()