def build_feed_dict(self):
if FLAGS.serialize_and_merge:
return self.build_feed_dict_with_serialize_and_merge()
_logger.info("Traversing trees.")
# The weaver is an object that can invoke LoomOps, and create a feed_dict.
weaver = self._loom.make_weaver()
# Recurse over each tree in the batch
for _ in six.moves.xrange(0, self.batch_size):
root = self.traverse_tree(self.get_input_tree(), weaver)
weaver.add_output(root)
# Now build the feed_dict, which contains both indices into the embedding
# tables, and indices for the gather nodes inserted by Loom.
if FLAGS.direct_feed_dict:
_logger.info("Calling build_feed_dict in direct mode.")
else:
_logger.info("Calling build_feed_dict with serialization.")
return weaver.build_feed_dict()
评论列表
文章目录