Home
last modified time | relevance | path

Searched refs:shape_from_pyval (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_client_backend_independent_test.py54 p0 = ops.Parameter(builder, 0, xla_client.shape_from_pyval(np.float32(0)))
56 xla_client.shape_from_pyval(np.zeros((4,), np.float32)))
87 p0 = ops.Parameter(builder0, 0, xla_client.shape_from_pyval(np.float32(0)))
89 xla_client.shape_from_pyval(np.zeros((4,), np.float32)))
94 p0 = ops.Parameter(builder1, 0, xla_client.shape_from_pyval(np.float32(0)))
96 xla_client.shape_from_pyval(np.zeros((4,), np.float32)))
109 xla_client.shape_from_pyval(np.array(
113 xla_client.shape_from_pyval(np.array(
H A Dxla_client_test.py141 p0 = ops.Parameter(builder, 0, xla_client.shape_from_pyval(np.float32(0)))
143 builder, 1, xla_client.shape_from_pyval(np.zeros((4,), np.float32)))
429 shape = xla_client.shape_from_pyval(arg0)
448 shape = xla_client.shape_from_pyval(arg0)
468 shape = xla_client.shape_from_pyval(arg0)
492 p0 = ops.Parameter(c, 0, xla_client.shape_from_pyval(arg0))
532 p0 = ops.Parameter(c, 0, xla_client.shape_from_pyval(arg0))
533 p1 = ops.Parameter(c, 1, xla_client.shape_from_pyval(arg1))
550 p1 = ops.Parameter(c, 1, xla_client.shape_from_pyval(arg1))
551 p0 = ops.Parameter(c, 0, xla_client.shape_from_pyval(arg0))
[all …]
H A Dxla_client.pyi63 def shape_from_pyval(pyval: Any) -> Any: ...
H A Dxla_client.py255 def shape_from_pyval(pyval): function