Home
last modified time | relevance | path

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

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/utils/
H A Djit_io_helper.cpp83 , gather_conf_(gather_conf) { in jit_io_helper_t()
189 assert(gather_conf_->vmm_tmp_idx_.has_value() in emu_gather()
197 host_->mov(gather_conf_->reg_tmp_, 0); in emu_gather()
198 host_->mov(gather_conf_->reg_tmp1_, src_reg); in emu_gather()
247 assert(gather_conf_->vmm_tmp_idx_.has_value() in emu_gather()
255 host_->mov(gather_conf_->reg_tmp_, 0); in emu_gather()
314 host_->mov(gather_conf_->reg_tmp_, 0); in emu_gather()
371 prepare_opmask(gather_conf_->simd_w_, gather_conf_->reg_tmp_, in prepare_full_mask()
372 gather_conf_->full_opmask_); in prepare_full_mask()
374 prepare_vmm_mask(gather_conf_->simd_w_, gather_conf_->simd_w_, in prepare_full_mask()
[all …]
H A Djit_io_helper.hpp196 const utils::optional_t<io_gather_conf_t> gather_conf_; member in dnnl::impl::cpu::x64::io::jit_io_helper_t
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/utils/
H A Djit_io_helper.cpp83 , gather_conf_(gather_conf) { in jit_io_helper_t()
189 assert(gather_conf_->vmm_tmp_idx_.has_value() in emu_gather()
197 host_->mov(gather_conf_->reg_tmp_, 0); in emu_gather()
198 host_->mov(gather_conf_->reg_tmp1_, src_reg); in emu_gather()
247 assert(gather_conf_->vmm_tmp_idx_.has_value() in emu_gather()
255 host_->mov(gather_conf_->reg_tmp_, 0); in emu_gather()
314 host_->mov(gather_conf_->reg_tmp_, 0); in emu_gather()
371 prepare_opmask(gather_conf_->simd_w_, gather_conf_->reg_tmp_, in prepare_full_mask()
372 gather_conf_->full_opmask_); in prepare_full_mask()
374 prepare_vmm_mask(gather_conf_->simd_w_, gather_conf_->simd_w_, in prepare_full_mask()
[all …]
H A Djit_io_helper.hpp196 const utils::optional_t<io_gather_conf_t> gather_conf_; member in dnnl::impl::cpu::x64::io::jit_io_helper_t