Searched refs:row_idx_vector (Results 1 – 1 of 1) sorted by relevance
373 std::vector<xla::XlaOp> row_idx_vector = {row_idx}; in operator ()() local374 auto active_elem = xla::DynamicSlice(included_iou, row_idx_vector, {1}); in operator ()()