Searched refs:_ymask (Results 1 – 1 of 1) sorted by relevance
253 def _ymask(length): function261 extra_terms = y0 + is_even * ym * _ymask(input_shape[-1])282 ], 0)) * _ymask(inner_dim)