Searched refs:all_reduce_spec (Results 1 – 1 of 1) sorted by relevance
812 all_reduce_spec=("pscpu/pscpu", 2, -1), argument857 "A tuple is expected for all-reduce spec: %r" % all_reduce_spec)869 if isinstance(all_reduce_spec, six.string_types):870 self._all_reduce_spec.append(AllReduceSpecTuple(all_reduce_spec, 1, -1))871 elif isinstance(all_reduce_spec, tuple):872 self._all_reduce_spec.append(validate_and_complete_spec(all_reduce_spec))873 elif isinstance(all_reduce_spec, list):875 validate_and_complete_spec(spec) for spec in all_reduce_spec