Searched refs:from_pyval (Results 1 – 3 of 3) sorted by relevance
415 arg_buffers = [xla_client.LocalBuffer.from_pyval(arg) for arg in arguments]446 arg_buffer = xla_client.LocalBuffer.from_pyval(arg)453 local_buffer = xla_client.LocalBuffer.from_pyval(t)460 local_buffer = xla_client.LocalBuffer.from_pyval(t)472 local_buffer = xla_client.LocalBuffer.from_pyval(t)486 local_buffer = xla_client.LocalBuffer.from_pyval(t)502 local_buffer = xla_client.LocalBuffer.from_pyval(pyval)1652 c.Infeed(xla_client.Shape.from_pyval(to_infeed[0]))1664 x = c.Infeed(xla_client.Shape.from_pyval(to_round_trip[0]))1674 xla_client.Shape.from_pyval(to_round_trip[0]))
416 def from_pyval(pyval, device=0, backend=None): member in LocalBuffer495 def from_pyval(pyval): member in Shape789 argument_shapes=[Shape.from_pyval(arg) for arg in arguments],875 return LocalBuffer.from_pyval(885 return LocalBuffer.from_pyval(arg, device, backend=self._backend)1063 Shape.from_pyval(value), name=name, parameter_num=parameter_num)
99 def from_pyval(pyval): member in Shape