Searched defs:dim_w (Results 1 – 5 of 5) sorted by relevance
29 int64_t dim_w = 1; in TORCH_META_FUNC() local77 int64_t dim_w = 1; in TORCH_META_FUNC() local113 int64_t dim_w = 3; in TORCH_META_FUNC() local179 int64_t dim_w = 3; in TORCH_META_FUNC() local214 int dim_w = 2; in reflection_pad2d_out_template() local272 int dim_w = 2; in reflection_pad2d_backward_out_template() local
710 SymDimVector dim_w(self_dim, 1); in prelu() local
279 int dim_w = 2; in reflection_pad2d_out_template() local371 int dim_w = 2; in reflection_pad2d_backward_out_template() local441 int64_t dim_w = 1; in TORCH_IMPL_FUNC() local498 int64_t dim_w = 1; in TORCH_IMPL_FUNC() local
52 int dim_n, dim_h, dim_w, dim_c; member64 int dim_h, dim_w, dim_i, dim_o; member
686 DimVector dim_w(input_ndim, 1); in qprelu_out_kernel() local