Searched refs:pfor_ops (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 62 pfor_ops = LazyLoader( variable
1175 output = pfor_ops.pfor(loop_fn, target_size,
1191 output = pfor_ops.for_loop(
1311 output = pfor_ops.pfor(loop_fn, target_row_size,
1327 output = pfor_ops.for_loop(loop_fn, target.dtype, target_row_size,
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 204 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument
216 self._pfor_ops = set(pfor_ops)
217 self._pfor_op_ids = set(x._id for x in pfor_ops)
394 pfor_ops=self._pfor_ops,
1267 pfor_ops, argument
1307 self._pfor_ops = set(pfor_ops)
1308 self._pfor_op_ids = set(x._id for x in pfor_ops)
1478 y, pfor_ops=self._pfor_ops,
1674 def pfor_ops(self): member in PFor
4187 pfor_ops=func.graph.get_operations(),
[all …]
|