Searched refs:shape_from_pyval (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_backend_independent_test.py | 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)))
91 p0 = ops.Parameter(builder0, 0, xla_client.shape_from_pyval(np.float32(0)))
93 xla_client.shape_from_pyval(np.zeros((4,), np.float32)))
98 p0 = ops.Parameter(builder1, 0, xla_client.shape_from_pyval(np.float32(0)))
100 xla_client.shape_from_pyval(np.zeros((4,), np.float32)))
113 xla_client.shape_from_pyval(np.array(
117 xla_client.shape_from_pyval(np.array(
|
| D | xla_client_test.py | 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)))
377 p0 = ops.Parameter(c, 0, xla_client.shape_from_pyval(arg0))
378 p1 = ops.Parameter(c, 1, xla_client.shape_from_pyval(arg1))
395 p1 = ops.Parameter(c, 1, xla_client.shape_from_pyval(arg1))
396 p0 = ops.Parameter(c, 0, xla_client.shape_from_pyval(arg0))
415 ops.Parameter(c, 0, xla_client.shape_from_pyval(NumpyArrayF32(0.))),
423 ops.Parameter(c, 0, xla_client.shape_from_pyval(NumpyArrayF32(0.))),
424 ops.Parameter(c, 1, xla_client.shape_from_pyval(NumpyArrayF32(0.))))
435 ops.Parameter(c, 0, xla_client.shape_from_pyval(NumpyArrayF32(0.))),
[all …]
|
| D | xla_client.py | 282 def shape_from_pyval(pyval): function
|