Searched refs:get_mask_for_dim (Results 1 – 1 of 1) sorted by relevance
491 auto get_mask_for_dim = [&](int64 dim, HloInstruction* start_index) { in PadWithValue() local525 mask = get_mask_for_dim(i, offsets[i]); in PadWithValue()529 get_mask_for_dim(i, offsets[i]))); in PadWithValue()