Searched refs:channels_out (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | TransposeConv2D.cpp | 491 uint32_t channels_out = getSizeOfDimension(filter, 0); in prepare() local
496 NN_RET_CHECK_EQ(channels_out, getSizeOfDimension(bias, 0)); in prepare()
500 NN_RET_CHECK_GT(channels_out, 0); in prepare()
514 output.dimensions = {batches, channels_out, outHeight, outWidth}; in prepare()
516 output.dimensions = {batches, outHeight, outWidth, channels_out}; in prepare()
|
| D | Conv2D.cpp | 515 uint32_t channels_out = getSizeOfDimension(filter, 0); in prepare() local
520 NN_RET_CHECK_EQ(channels_out, getSizeOfDimension(bias, 0)); in prepare()
524 NN_RET_CHECK_GT(channels_out, 0); in prepare()
543 output.dimensions = {batches, channels_out, outHeight, outWidth}; in prepare()
545 output.dimensions = {batches, outHeight, outWidth, channels_out}; in prepare()
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 371 uint32_t channels_out = getSizeOfDimension(filter, 3); in depthwiseConvPrepare() local
379 NN_OPS_CHECK(depth_multiplier * channels_in == channels_out); in depthwiseConvPrepare()
393 output->dimensions = {batches, outHeight, outWidth, channels_out}; in depthwiseConvPrepare()
889 uint32_t channels_out = getSizeOfDimension(filter, 0); in groupedConvPrepare() local
907 output->dimensions = {batches, outHeight, outWidth, channels_out}; in groupedConvPrepare()
|