Home
last modified time | relevance | path

Searched refs:dst_dim_idx (Results 1 – 1 of 1) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/
H A Dkernel_builder.cpp3443 int dst_dim_idx = 0; in create_reduce_stmt() local
3446 dst2src[dst_dim_idx] = i; in create_reduce_stmt()
3447 dst_dim_idx++; in create_reduce_stmt()
3450 ir_assert(dst_dim_idx == dst.ndims()) << "Incompatible reduction mask."; in create_reduce_stmt()