Searched refs:dim_assignment_t (Results 1 – 3 of 3) sorted by relevance
616 layout_t dim_assignment_t::map(const layout_t &layout) const { in map()641 dim_assignment_t to_3d(old_ndims, new_ndims); in normalize_conv_spatial()669 dim_assignment_t a(layout.ndims(), layout.ndims() - 1); in remove_size_1_dimension()
1593 class dim_assignment_t { class1595 dim_assignment_t() = default;1597 dim_assignment_t(int old_ndims, int new_ndims) in dim_assignment_t() function in dnnl::impl::gpu::jit::dim_assignment_t
1803 dim_assignment_t to_ab(src_.ndims(), 2); in emit()