Searched refs:channelOut (Results 1 – 1 of 1) sorted by relevance
293 RandomVariable channelOut = in depthwiseConv2DExplicitConstructor() local295 op->inputs[1]->dimensions = {1, RandomVariableType::FREE, RandomVariableType::FREE, channelOut}; in depthwiseConv2DExplicitConstructor()298 op->inputs[2]->dimensions = {channelOut}; in depthwiseConv2DExplicitConstructor()305 op->outputs[0]->dimensions[channelIndex] = channelOut; in depthwiseConv2DExplicitConstructor()346 RandomVariable channelOut = in depthwiseConv2DImplicitConstructor() local348 op->inputs[1]->dimensions = {1, RandomVariableType::FREE, RandomVariableType::FREE, channelOut}; in depthwiseConv2DImplicitConstructor()351 op->inputs[2]->dimensions = {channelOut}; in depthwiseConv2DImplicitConstructor()358 op->outputs[0]->dimensions[channelIndex] = channelOut; in depthwiseConv2DImplicitConstructor()