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 D | jit_io_helper.cpp | 83 , 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 D | jit_io_helper.hpp | 196 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 D | jit_io_helper.cpp | 83 , 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 D | jit_io_helper.hpp | 196 const utils::optional_t<io_gather_conf_t> gather_conf_; member in dnnl::impl::cpu::x64::io::jit_io_helper_t
|