Searched refs:vectorized_compute (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | xla_control_flow_ops_test.py | 72 def vectorized_compute(x): function
76 vectorized_compute, inputs=[array_ops.ones((10, 5, 3))])
85 def vectorized_compute(x): function
88 result = vectorized_compute(array_ops.ones((10, 5, 3)))
96 def vectorized_compute(x, i): function
105 lambda i, y: (i + 1, y + vectorized_compute(x, i)),
|