class_docstr: Class to bind an optimizer, model, strategy, and training step function.
class_docstr_tok: ["Class","to","bind","an","optimizer",",","model",",","strategy",",","and","training","step","function","."]
docstr: Sum across processors.
docstr_tok: ["Sum","across","processors","."]
code: def psum(self, x, axis=None):
      """Sum across processors."""
      return self.strategy.reduce(
          tf.distribute.ReduceOp.SUM, x, axis=axis
      )
  
file_path: ddsp/ddsp/training/trainers.py
imports: import time
  from absl import logging
  from ddsp.training import train_util
  import gin
  import tensorflow.compat.v2
label_desc: Use image node_id 6 for calling the Trainer obj's underlying member method code with example usage: obj.psum(x, axis) and returns: self