Searched refs:expanded_feature_iota (Results 1 – 1 of 1) sorted by relevance
111 xla::XlaOp expanded_feature_iota = xla::Iota( in CreateExpandedFilterMask() local116 expanded_feature_iota = in CreateExpandedFilterMask()117 xla::Div(expanded_feature_iota, in CreateExpandedFilterMask()123 return xla::Eq(expanded_feature_iota, input_feature_iota); in CreateExpandedFilterMask()