Searched defs:weight_dim (Results 1 – 4 of 4) sorted by relevance
146 int64_t weight_dim = weight.ndimension(); in check_shape_forward() local
77 const auto weight_dim = weight.dim(); in linear() local
660 int64_t weight_dim = weight_sizes.size(); in check_shape_forward() local
65 int64_t weight_dim = weight_sizes.size(); in check_shape_forward() local