Home
last modified time | relevance | path

Searched refs:output_tensor_device_ids (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/debug/lib/
Ddumping_callback.py458 output_tensor_device_ids, argument
485 assert len(tensors) == len(output_tensor_device_ids)
493 output_tensor_device_ids=output_tensor_device_ids,
507 output_tensor_device_ids=output_tensor_device_ids,
599 output_tensor_device_ids = [writer.RegisterDeviceAndGetId(output.device)
602 outputs, op_type, input_ids, output_tensor_device_ids,
Ddebug_events_monitors_test.py89 self.assertLen(execution.output_tensor_device_ids, 1)
140 self.assertLen(execution.output_tensor_device_ids, 2)
244 1234, 1, "FooOp", output_tensor_device_ids=[0, 1])
288 1234, 1, "BarOp", output_tensor_device_ids=[0, 1])
327 1234, 1, "BarOp", output_tensor_device_ids=[0, 1])
363 output_tensor_device_ids=[0, 1])
476 i * 1000, 1, "FooOp", output_tensor_device_ids=[0, 1])
Ddebug_events_reader.py373 output_tensor_device_ids=None): argument
376 self._output_tensor_device_ids = _tuple_or_none(output_tensor_device_ids)
383 def output_tensor_device_ids(self): member in ExecutionDigest
390 "output_tensor_device_ids": self.output_tensor_device_ids,
440 output_tensor_device_ids=execution_digest.output_tensor_device_ids)
887 output_tensor_device_ids=(debug_event.execution.output_tensor_device_ids
Ddebug_events_writer_test.py691 1234, 5678, "FooOp", output_tensor_device_ids=None)
700 1234, 5678, "FooOp", output_tensor_device_ids=[1357, 2468])
709 1234, 5678, "FooOp", output_tensor_device_ids=[1357])
735 1234, 5678, "FooOp", output_tensor_device_ids=[1357])
Ddumping_callback_test.py165 self.assertLen(execution.output_tensor_device_ids, 1)
167 reader.device_name_by_id(execution.output_tensor_device_ids[0]),
501 self.assertLen(executions[0].output_tensor_device_ids, 1)
503 reader.device_name_by_id(executions[0].output_tensor_device_ids[0]),
/external/tensorflow/tensorflow/core/protobuf/
Ddebug_event.proto265 repeated int32 output_tensor_device_ids = 9; field