Searched refs:pfor_ops (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 60 pfor_ops = LazyLoader( variable
1216 output = pfor_ops.pfor(loop_fn, target_size,
1229 output = pfor_ops.for_loop(
1363 output = pfor_ops.pfor(loop_fn, target_row_size,
1376 output = pfor_ops.for_loop(loop_fn, target.dtype, target_row_size,
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 211 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
223 self._pfor_ops = set(pfor_ops)
224 self._pfor_op_ids = set(x._id for x in pfor_ops)
401 pfor_ops=self._pfor_ops,
1275 pfor_ops, argument
1317 self._pfor_ops = set(pfor_ops)
1318 self._pfor_op_ids = set(x._id for x in pfor_ops)
1489 y, pfor_ops=self._pfor_ops,
1692 def pfor_ops(self): member in PFor
4504 pfor_ops=func.graph.get_operations(),
[all …]
|