Searched refs:_wrap_shape (Results 1 – 1 of 1) sorted by relevance
427 return _wrap_shape(self.c_buffer.shape())631 def _wrap_shape(shape_info): function635 shapes = tuple(_wrap_shape(subshape_info) for subshape_info in dims)643 return ProgramShape([_wrap_shape(arg) for arg in arg_shapes],644 _wrap_shape(result_shape))767 result_shape = _wrap_shape(self.computation.GetReturnValueShape())798 return _wrap_shape(self._c_computation.GetReturnValueShape())1155 return _wrap_shape(self._client.GetShape(operand))1158 return _wrap_shape(self._client.GetReturnValueShape())