Lines Matching defs:kz
214 /* Find index of greatest value of abs_dir and put that as kz. */
215 nir_ssa_def *kz = nir_bcsel(
219 nir_ssa_def *kx = nir_imod(b, nir_iadd_imm(b, kz, 1), nir_imm_int(b, 3));
221 nir_ssa_def *k_indices[3] = {kx, ky, kz};
226 k = nir_bcsel(b, nir_flt(b, nir_vector_extract(b, dir, kz), nir_imm_float(b, 0.0f)),
231 kz = nir_channel(b, k, 2);
234 nir_ssa_def *sz = nir_frcp(b, nir_vector_extract(b, dir, kz));
245 nir_fsub(b, nir_vector_extract(b, v_a, kx), nir_fmul(b, sx, nir_vector_extract(b, v_a, kz)));
247 nir_fsub(b, nir_vector_extract(b, v_a, ky), nir_fmul(b, sy, nir_vector_extract(b, v_a, kz)));
249 nir_fsub(b, nir_vector_extract(b, v_b, kx), nir_fmul(b, sx, nir_vector_extract(b, v_b, kz)));
251 nir_fsub(b, nir_vector_extract(b, v_b, ky), nir_fmul(b, sy, nir_vector_extract(b, v_b, kz)));
253 nir_fsub(b, nir_vector_extract(b, v_c, kx), nir_fmul(b, sx, nir_vector_extract(b, v_c, kz)));
255 nir_fsub(b, nir_vector_extract(b, v_c, ky), nir_fmul(b, sy, nir_vector_extract(b, v_c, kz)));
317 nir_ssa_def *az = nir_fmul(b, sz, nir_vector_extract(b, v_a, kz));
318 nir_ssa_def *bz = nir_fmul(b, sz, nir_vector_extract(b, v_b, kz));
319 nir_ssa_def *cz = nir_fmul(b, sz, nir_vector_extract(b, v_c, kz));