Searched refs:fixed_dim (Results 1 – 1 of 1) sorted by relevance
3398 def _expand_mask(mask_t, input_t, fixed_dim=1): argument3404 multiples = [1] * fixed_dim + input_t.shape.as_list()[fixed_dim:]3471 _expand_mask(mask, outputs, fixed_dim=2),