Searched defs:weight_in (Results 1 – 1 of 1) sorted by relevance
105 at::Tensor rearrange_weights_dw(const Tensor& weight_in) { in rearrange_weights_dw()184 at::Tensor rearrange_weights_2d(const Tensor& weight_in, bool tconv) { in rearrange_weights_2d()249 const at::Tensor& weight_in, in rearrange_bias()