Home
last modified time | relevance | path

Searched refs:xla_client (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_backend_independent_test.py26 from tensorflow.compiler.xla.python import xla_client
35 ops = xla_client.ops
42 xla_client.Shape.array_shape(xla_client.PrimitiveType.F32, [-2, 4])
46 xla_client.Shape.array_shape(xla_client.PrimitiveType.F32, [2, 4], [3])
50 xla_client.Shape.array_shape(xla_client.PrimitiveType.F32, [2, 4],
57 builder = xla_client.XlaBuilder("acomputation")
58 p0 = ops.Parameter(builder, 0, xla_client.shape_from_pyval(np.float32(0)))
60 xla_client.shape_from_pyval(np.zeros((4,), np.float32)))
82 hlo_dot_graph = xla_client._xla.hlo_module_to_dot_graph(
90 builder0 = xla_client.XlaBuilder("computation0")
[all …]
Dxla_client_test.py28 from tensorflow.compiler.xla.python import xla_client
36 bfloat16 = xla_client.bfloat16
37 ops = xla_client.ops
72 return xla_client.XlaBuilder(name)
76 return xla_client.execute_with_python_values(
127 builder = xla_client.XlaBuilder("acomputation")
128 p0 = ops.Parameter(builder, 0, xla_client.shape_from_pyval(np.float32(0)))
130 builder, 1, xla_client.shape_from_pyval(np.zeros((4,), np.float32)))
148 properties = xla_client._xla.hlo_module_cost_analysis(
205 ops.Iota(c, xla_client.PrimitiveType.F32, 10)
[all …]
DBUILD18 name = "xla_client",
19 srcs = ["xla_client.py"],
55 ":xla_client",
68 ":xla_client",
86 ":xla_client",
106 ":xla_client",
Dcustom_call_for_test.pyx5 # it via the xla_client SWIG API.
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc76 LocalClient* xla_client, bool asynchronous) { in BuildLocalDeviceStates() argument
78 for (int i = 0; i < xla_client->device_count(); ++i) { in BuildLocalDeviceStates()
80 xla_client->backend().stream_executor(i).ValueOrDie(); in BuildLocalDeviceStates()
82 executor, xla_client, LocalDeviceState::kComputeSynchronized, in BuildLocalDeviceStates()
316 TF_ASSIGN_OR_RETURN(LocalClient * xla_client, GetGpuXlaClient()); in GetGpuClient()
319 BuildLocalDeviceStates(xla_client, asynchronous)); in GetGpuClient()
337 kGpuName, xla_client, std::move(devices), in GetGpuClient()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.py25 from tensorflow.compiler.xla.python import xla_client # pylint: disable=unused-import
DBUILD84 "//tensorflow/compiler/xla/python:xla_client",