Searched defs:dim_h (Results 1 – 3 of 3) sorted by relevance
114 int64_t dim_h = 2; in TORCH_META_FUNC() local180 int64_t dim_h = 2; in TORCH_META_FUNC() local215 int dim_h = 1; in reflection_pad2d_out_template() local273 int dim_h = 1; in reflection_pad2d_backward_out_template() local
52 int dim_n, dim_h, dim_w, dim_c; member64 int dim_h, dim_w, dim_i, dim_o; member
278 int dim_h = 1; in reflection_pad2d_out_template() local370 int dim_h = 1; in reflection_pad2d_backward_out_template() local