Searched refs:_array_aval_from_xla_shape (Results 1 – 1 of 1) sorted by relevance
943 avals = [_array_aval_from_xla_shape(c.get_shape(x)) for x in xla_args]973 def _array_aval_from_xla_shape(xla_shape): function