Searched defs:w_sizes (Results 1 – 4 of 4) sorted by relevance
79 const auto w_sizes = weight.sizes(); in slow_conv2d_shape_check() local96 const auto w_sizes = weight.sizes(); in slow_conv2d_shape_check() local118 const auto w_sizes = weight->sizes(); in new_view_weight_MM2d() local
375 auto w_sizes = weight.sizes(); in conv_depthwise2d_forward_out() local
42 SizesArrayRef w_sizes, in conv2d_impl()
97 const IntArrayRef w_sizes = weight.sizes(); in pack_weights() local