Searched refs:bmnk_kinds (Results 1 – 3 of 3) sorted by relevance
/dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
H A D | gemm_schedule.cpp | 25 const std::vector<bmnk_kind_t> &bmnk_kinds, const view_t &view) const { in map_to_bmnk() argument 27 return map_to_bmnk(abc_kind, bmnk_kinds, layout); in map_to_bmnk() 31 const std::vector<bmnk_kind_t> &bmnk_kinds, in map_to_bmnk() argument 37 for (int i = 0; i < int(bmnk_kinds.size()); i++) { in map_to_bmnk() 38 if (bmnk_kinds[i] == b_bmnk_kind) { in map_to_bmnk() 46 return layout_t(layout.type(), int(bmnk_kinds.size()), 0, blocks); in map_to_bmnk() 60 const std::vector<bmnk_kind_t> &bmnk_kinds, in map_from_bmnk() argument 74 auto &bmnk_blocks = tmp_blocks[static_cast<int>(bmnk_kinds[b.dim_idx])]; in map_from_bmnk() 79 for (auto bmnk_kind : bmnk_kinds) { in map_from_bmnk()
|
H A D | gemm_schedule.hpp | 52 bmnk_mapper_t(const object_map_t<expr_t, bmnk_kind_t> &bmnk_kinds) in bmnk_mapper_t() argument 53 : bmnk_kinds_(bmnk_kinds) {} in bmnk_mapper_t() 91 const std::vector<bmnk_kind_t> &bmnk_kinds, 95 const std::vector<bmnk_kind_t> &bmnk_kinds, 134 const std::vector<bmnk_kind_t> &bmnk_kinds,
|
H A D | kernel_builder.cpp | 4670 std::vector<bmnk_kind_t> bmnk_kinds; in convert_to_fma_friendly_layout() local 4672 bmnk_kinds.push_back(bmnk_kind_t::m); in convert_to_fma_friendly_layout() 4673 bmnk_kinds.push_back(bmnk_kind_t::k); in convert_to_fma_friendly_layout() 4675 bmnk_kinds.push_back(bmnk_kind_t::k); in convert_to_fma_friendly_layout() 4676 bmnk_kinds.push_back(bmnk_kind_t::n); in convert_to_fma_friendly_layout() 4679 auto bmnk_layout = bmnk_mapper.map_to_bmnk(abc_kind, bmnk_kinds, layout); in convert_to_fma_friendly_layout() 4691 = from_bmnk_mapper.map_from_bmnk(abc_kind, bmnk_kinds, dpas_layout); in convert_to_fma_friendly_layout()
|