Searched refs:active_elem (Results 1 – 1 of 1) sorted by relevance
374 auto active_elem = xla::DynamicSlice(included_iou, row_idx_vector, {1}); in operator ()() local375 active_elem = xla::Reshape(active_elem, {}); in operator ()()378 active_elem, num_outputs_so_far + xla::ConstantR0<int32>(builder, 1), in operator ()()391 included_iou = xla::Select(xla::Broadcast(active_elem, {num_boxes}), in operator ()()