Searched refs:GenerateBinaryMask (Results 1 – 1 of 1) sorted by relevance
315 HloInstruction* GenerateBinaryMask( in GenerateBinaryMask() function490 GenerateBinaryMask(reshape, input_dim, output_dims, output_dynamic_dims, in RewriteDynamicReshapeSplitInput()669 GenerateBinaryMask(reshape, output_dim, input_dims, input_dynamic_dims, in RewriteDynamicReshapeCombineInput()