Searched defs:new_state_size (Results 1 – 3 of 3) sorted by relevance
126 auto new_state_size = new_state.numel(); in set_state() local
159 auto new_state_size = new_state.numel(); in set_state() local
355 auto new_state_size = new_state.numel(); in set_state() local