Searched defs:gO_sizes (Results 1 – 2 of 2) sorted by relevance
481 auto gO_sizes = grad_output.sizes(); in conv_depthwise2d_backward_out() local607 auto gO_sizes = grad_output.sizes(); in conv_depthwise2d_grad_weight_out() local
90 const auto gO_sizes = grad_output.sizes(); in slow_conv2d_shape_check() local