Searched defs:iter_mask (Results 1 – 6 of 6) sorted by relevance
128 TensorIterator & iter_mask, in _fake_quant_per_channel_cachemask_cuda_helper()184 TensorIterator &iter, TensorIterator &iter_mask, int64_t quant_min, int64_t quant_max) { in fake_quant_per_channel_cachemask_cuda()
95 TensorIterator iter_mask = TensorIteratorConfig() in fake_quantize_per_channel_affine_cachemask() local
220 uint32_t iter_mask = 3; in setup_parms() local
3434 unsigned iter_mask = wr_mask; in nir_instr_xfb_write_mask() local
2643 TensorIterator& iter_mask,2703 TensorIterator& iter_mask,
11638 Temp iter_mask = bld.copy(bld.def(v1), Operand::c32(ps_iter_mask)); in overwrite_samplemask_arg() local