Searched refs:input_shape_binary_mask (Results 1 – 1 of 1) sorted by relevance
489 HloInstruction* input_shape_binary_mask = in RewriteDynamicReshapeSplitInput() local492 if (input_shape_binary_mask == nullptr) { in RewriteDynamicReshapeSplitInput()523 mask_input_shape, input_shape_binary_mask, zero, cumsum_window, add)); in RewriteDynamicReshapeSplitInput()