Home
last modified time | relevance | path

Searched refs:pyval (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py84 def buffer_from_pyval(self, pyval, device=0): argument
132 def buffer_from_pyval(self, pyval, device=0): argument
133 return c_api.LocalShapedBuffer.FromLiteral(pyval, None, self.client, device)
169 def buffer_from_pyval(self, pyval, device=0): argument
173 return xrt_api.XrtAllocation.FromLiteral(pyval,
416 def from_pyval(pyval, device=0, backend=None): argument
419 pyval = require_numpy_array_layout(pyval)
420 cbuf = backend.buffer_from_pyval(pyval, device)
495 def from_pyval(pyval): argument
496 def convert(pyval): argument
[all …]
Dxla_client_test.py501 pyval = np.array([[1., 2.]], np.float32)
502 local_buffer = xla_client.LocalBuffer.from_pyval(pyval)
/external/tensorflow/tensorflow/compiler/xla/python_api/
Dxla_shape.py99 def from_pyval(pyval): argument
100 return CreateShapeFromNumpy(pyval)