Lines Matching refs:datum
201 def has_inf_or_nan(datum, tensor):
210 datum: (`DebugTensorDatum`) Datum metadata.
218 _ = datum # Datum metadata is unused in this predicate.
563 datum = self._dump_file_name_to_datum(root, f)
564 self._dump_tensor_data[device_name].append(datum)
565 self._debug_watches[device_name][datum.node_name][
566 datum.output_slot].add(datum.debug_op)
625 """Create maps from tensor watch keys to datum and to timestamps.
638 for datum in self._dump_tensor_data[device_name]:
639 if datum.watch_key not in self._watch_key_to_devices:
640 self._watch_key_to_devices[datum.watch_key] = {device_name}
642 self._watch_key_to_devices[datum.watch_key].add(device_name)
644 if datum.watch_key not in self._watch_key_to_datum[device_name]:
645 self._watch_key_to_datum[device_name][datum.watch_key] = [datum]
646 self._watch_key_to_rel_time[device_name][datum.watch_key] = [
647 datum.timestamp - self._t0]
648 self._watch_key_to_dump_size_bytes[device_name][datum.watch_key] = [
649 datum.dump_size_bytes]
651 self._watch_key_to_datum[device_name][datum.watch_key].append(datum)
652 self._watch_key_to_rel_time[device_name][datum.watch_key].append(
653 datum.timestamp - self._t0)
654 self._watch_key_to_dump_size_bytes[device_name][datum.watch_key].append(
655 datum.dump_size_bytes)
839 for datum in self._dump_tensor_data[device_name]:
840 if datum.node_name not in debug_graph.node_inputs:
842 "device %s." % (datum.node_name, device_name))
861 for i, datum in enumerate(self._dump_tensor_data[device_name]):
862 node = datum.node_name
863 slot = datum.output_slot
869 device_name, pending_inputs[node], datum.timestamp, start_i=i + 1):
873 (node, datum.timestamp, repr(pending_inputs[node])))
907 for datum in self._dump_tensor_data[device_name][start_i:]:
908 if datum.timestamp > timestamp:
910 if (datum.timestamp == timestamp and
911 (datum.node_name, datum.output_slot) in pending):
912 pending.remove((datum.node_name, datum.output_slot))
1443 for datum in self._dump_tensor_data[device]:
1444 if predicate(datum, datum.get_tensor()):
1445 matched_data.append(datum)
1483 return [datum.file_path for datum in
1511 return [datum.get_tensor() for datum in
