Searched refs:num_outputs_so_far (Results 1 – 1 of 1) sorted by relevance
345 xla::XlaOp num_outputs_so_far = values[1]; in operator ()() local347 num_outputs_so_far, xla::ConstantR0<int32>(cond_builder, output_size)); in operator ()()364 auto num_outputs_so_far = values[1]; in operator ()() local377 num_outputs_so_far = xla::Select( in operator ()()378 active_elem, num_outputs_so_far + xla::ConstantR0<int32>(builder, 1), in operator ()()379 num_outputs_so_far); in operator ()()394 return std::vector<xla::XlaOp>{row_idx, num_outputs_so_far, iou_mask, in operator ()()