Searched refs:tensor_fetches (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 1468 def _flush_tensor_values_cache(self, tensor_fetches, op_fetches, on_tpu, argument
1484 [tensor.op for tensor in tensor_fetches]):
1487 return control_flow_ops.tuple(tensor_fetches,
1490 def _process_tensor_fetches(self, tensor_fetches): argument
1493 if tensor_fetches is None:
1496 if not isinstance(tensor_fetches, (list, tuple)):
1497 tensor_fetches = [tensor_fetches]
1498 elif not tensor_fetches:
1502 for fetch in tensor_fetches:
1727 tensor_fetches, argument
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | wrap_function.py | 287 tensor_fetches = []
311 tensor_fetches.append(decoded)
316 tensor_fetches.append(fetch)
325 tensor_fetches = nest.flatten(tensor_fetches, expand_composites=True)
327 for f in (flat_feeds + tensor_fetches + operation_fetches):
335 operation_fetches + tensor_fetches,
343 pruned_graph.outputs.extend(lift_map[x] for x in tensor_fetches)
|