Home
last modified time | relevance | path

Searched refs:count_mask (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtopk_ops.cc191 int32 count_mask = next_power_of_two - 1; in CreateTopKUnique() local
192 xla::XlaOp mask_r0 = xla::ConstantR0(builder, count_mask); in CreateTopKUnique()
270 uint32 count_mask = ~(next_power_of_two - 1); in CreateMakeUnique() local
271 xla::XlaOp count_mask_r0 = xla::ConstantR0(builder, count_mask); in CreateMakeUnique()
/external/webp/src/dsp/
Dlossless.c350 const int count_mask = pixels_per_byte - 1; \
360 if ((x & count_mask) == 0) packed_pixels = GET_INDEX(*src++); \
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_state_shaders.c1754 unsigned count_mask = (1 << count) - 1; in si_shader_selector_key_vs() local
1755 unsigned fix = elts->fix_fetch_always & count_mask; in si_shader_selector_key_vs()
1756 unsigned opencode = elts->fix_fetch_opencode & count_mask; in si_shader_selector_key_vs()
1759 uint32_t mask = elts->fix_fetch_unaligned & count_mask; in si_shader_selector_key_vs()