Method export_placement

tensorflow/python/grappler/controller.py:123–131

Annotate the placement onto the specified metagraph. Args: metagraph: the metagraph to annotate with the placement.

(self, metagraph)

    raise NotImplementedError

  def export_placement(self, metagraph):
    """Annotate the placement onto the specified metagraph.

    Args:
      metagraph: the metagraph to annotate with the placement.
    """
    for node in metagraph.graph_def.node:
      if node.name in self.important_op_names:
        node.device = self.get_node_by_name(node.name).device

  # Get the nodes in the immediate fanin of node.
  # Beware: this doesn't take into account the nodes that may be skipped