Searched refs:_get_shape (Results 1 – 3 of 3) sorted by relevance
535 def _get_shape(x): function539 static_first_dims = [_get_shape(elem) for elem in flat_elems]
997 t0_shape = _get_shape(t0)1006 t1_shape = _get_shape(t1)1050 def _get_shape(tensor): function
548 def _get_shape(tensor): function556 return nest.map_structure(_get_shape, nested)