Searched defs:norms (Results 1 – 11 of 11) sorted by relevance
41 std::vector<Tensor> norms; variable53 std::vector<Tensor> norms; variable
104 accscalar_t* __restrict__ norms, in weight_norm_fwd_first_dim_kernel()158 accscalar_t* __restrict__ norms, in weight_norm_fwd_last_dim_kernel()354 auto norms = at::empty_strided(g.sizes(), g.strides(), g.options().dtype(AccType)); in weight_norm_cuda() local
102 Norms norms; variable
193 Norm *norms; variable
29 norms = torch.norm(drs, dim=1).reshape(-1, 1) variable
142 auto norms = saved_norms.to(saved_g.scalar_type()); in _weight_norm_differentiable_backward() local
35 norms = init_ops.truncated_normal_initializer variable
286 FrobeniusNorms norms; in ComputeFrobeniusNorms() local
870 void GrAAConvexTessellator::Ring::init(const SkTDArray<SkVector>& norms, in init()