Searched refs:channels_out (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | TransposeConv2D.cpp | 464 uint32_t channels_out = getSizeOfDimension(filter, 0); in prepare() local
469 NN_RET_CHECK_EQ(channels_out, getSizeOfDimension(bias, 0)); in prepare()
473 NN_RET_CHECK_GT(channels_out, 0u); in prepare()
487 output.dimensions = {batches, channels_out, outHeight, outWidth}; in prepare()
489 output.dimensions = {batches, outHeight, outWidth, channels_out}; in prepare()
|
| D | Conv2D.cpp | 558 uint32_t channels_out = getSizeOfDimension(filter, 0); in prepare() local
563 NN_RET_CHECK_EQ(channels_out, getSizeOfDimension(bias, 0)); in prepare()
567 NN_RET_CHECK_GT(channels_out, 0u); in prepare()
586 output.dimensions = {batches, channels_out, outHeight, outWidth}; in prepare()
588 output.dimensions = {batches, outHeight, outWidth, channels_out}; in prepare()
|
| D | DepthwiseConv2D.cpp | 446 uint32_t channels_out = getSizeOfDimension(filter, 3); in prepare() local
450 NN_OPS_CHECK(param.depth_multiplier * channels_in == channels_out); in prepare()
468 output.dimensions = {batches, channels_out, outHeight, outWidth}; in prepare()
470 output.dimensions = {batches, outHeight, outWidth, channels_out}; in prepare()
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 641 uint32_t channels_out = getSizeOfDimension(filter, 0); in groupedConvPrepare() local
659 output->dimensions = {batches, outHeight, outWidth, channels_out}; in groupedConvPrepare()
|