Searched refs:input_bool (Results 1 – 2 of 2) sorted by relevance
258 auto input_bool = constant_as<bool>(n->input()); in inlineIf() local259 AT_ASSERT(input_bool); in inlineIf()264 *input_bool); in inlineIf()265 size_t block_index = *input_bool ? 0 : 1; in inlineIf()
201 repack_invocations_in_workgroup(nir_builder *b, nir_def **input_bool, in repack_invocations_in_workgroup() argument219 assert(input_bool[i]->bit_size == 1); in repack_invocations_in_workgroup()221 input_mask[i] = nir_ballot(b, 1, wave_size, input_bool[i]); in repack_invocations_in_workgroup()