Searched refs:_get_shape (Results 1 – 4 of 4) sorted by relevance
40 def _get_shape(tensor): function75 mapped_conditioning, _get_shape(tensor))
100 def _get_shape(tensor): function104 base_shape = tensor_shape.TensorShape(_get_shape(tensor_list[0]))106 base_shape.assert_is_compatible_with(_get_shape(tensor))
380 t0_shape = _get_shape(t0)389 t1_shape = _get_shape(t1)432 def _get_shape(tensor): function
501 def _get_shape(tensor): function509 return nest.map_structure(_get_shape, nested)