Searched defs:grad_dim (Results 1 – 1 of 1) sorted by relevance
108 int64_t grad_dim = grad.dim() > 0 ? grad.dim() : 1; in TORCH_META_FUNC() local140 int64_t grad_dim = grad.dim() > 0 ? grad.dim() : 1; in TORCH_META_FUNC() local