Searched refs:wei_view (Results 1 – 2 of 2) sorted by relevance
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
| H A D | kernel_builder.hpp | 51 view_t &wei_view, view_t &dst_view, expr_t &src_buf,
54 view_t &wei_view, view_t &src_view, expr_t &dst_buf,
57 view_t &dst_view, view_t &wei_view, view_t &bia_view,
|
| H A D | kernel_builder.cpp | 6040 wei_view.set_vdim(g, cfg_.g); in init_fwd()
6041 wei_view.set_vdim(oc, cfg_.oc); in init_fwd()
6048 wei_view.set_tdim(2, ic); in init_fwd()
6049 wei_view.set_tdim(3, kd); in init_fwd()
6050 wei_view.set_tdim(4, kh); in init_fwd()
6217 wei_view.set_tdim(0, oc); in init_bwd_d()
6219 wei_view.set_tdim(2, kd); in init_bwd_d()
6220 wei_view.set_tdim(3, kh); in init_bwd_d()
6221 wei_view.set_tdim(4, kw); in init_bwd_d()
6388 wei_view.set_tdim(2, kd); in init_bwd_w()
[all …]
|