Searched defs:grad_weight_t (Results 1 – 4 of 4) sorted by relevance
212 auto grad_weight_t = at::empty(weight->sizes(), weight->options()); in miopen_batch_norm_backward() local
1085 auto grad_weight_t = at::empty(weight_size, grad_output_contig->options(), memory_format); in miopen_depthwise_convolution_backward_weight() local1138 auto grad_weight_t = at::empty(weight_size, grad_output_contig->options(), memory_format); in miopen_convolution_backward_weight() local
329 auto grad_weight_t = at::empty(weight->sizes(), weight->options()); in cudnn_batch_norm_backward() local
525 auto grad_weight_t = in cudnn_convolution_backward_weight() local