Searched refs:weights_CHout_dim (Results 1 – 2 of 2) sorted by relevance
43 constexpr unsigned int weights_CHout_dim = 0u; variable201 const unsigned int weights_out_ch = weights.shape()[weights_CHout_dim]; in conv3d()
1441 constexpr unsigned int weights_CHout_dim = 0u; in compute_conv3d_shape() local1488 output_shape.set(channel_dim, weights[weights_CHout_dim]); in compute_conv3d_shape()