Searched refs:_get_const_output_shape (Results 1 – 1 of 1) sorted by relevance
30 def _get_const_output_shape(input_shape, dim): function171 output_shape = _get_const_output_shape(input_shape, dim=1)206 output_shape = _get_const_output_shape(input_shape, dim=1)227 output_shape = _get_const_output_shape(input_shape, dim=2)