Searched refs:tensor_fetches (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 1318 def _flush_tensor_values_cache(self, graph, tensor_fetches, op_fetches, argument
1334 [tensor.op for tensor in tensor_fetches]):
1340 return control_flow_ops.tuple(tensor_fetches,
1343 def _process_tensor_fetches(self, tensor_fetches): argument
1346 if tensor_fetches is None:
1349 if not isinstance(tensor_fetches, (list, tuple)):
1350 tensor_fetches = [tensor_fetches]
1351 elif not tensor_fetches:
1355 for fetch in tensor_fetches:
1411 tensor_fetches, argument
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | wrap_function.py | 160 tensor_fetches = []
164 tensor_fetches.append(f)
178 if tensor_fetches:
179 identity_fetches = array_ops.identity_n(tensor_fetches)
187 tensor_fetches, identity_fetches):
191 if not tensor_fetches:
|