Searched refs:input_feature_iota (Results 1 – 1 of 1) sorted by relevance
109 xla::XlaOp input_feature_iota = xla::Iota( in CreateExpandedFilterMask() local123 return xla::Eq(expanded_feature_iota, input_feature_iota); in CreateExpandedFilterMask()