(self, input_name, input_version)
| 884 | return sum_ops, sum_op_output |
| 885 | |
| 886 | def _MakeSumOps(self, input_name, input_version): |
| 887 | generators = self.gradient_generators[input_name][input_version] |
| 888 | out_base_name = self._GetSumOpOutputName(generators, input_name) |
| 889 | types = list(set(type(x) for x in generators)) |
| 890 | assert(len(types) == 1) |
| 891 | if types[0] is GradGenMeta: |
| 892 | sum_ops, g = self._MakeDenseSumOps(generators, out_base_name) |
| 893 | else: |
| 894 | assert(types[0] is SparseGradGenMeta) |
| 895 | sum_ops, g = self._MakeSparseSumOps(generators, out_base_name) |
| 896 | self._SetSumOpsDeviceOption(sum_ops, generators) |
| 897 | return sum_ops, g |
| 898 | |
| 899 | def _VerifyGradientGenerators(self, generator): |
| 900 | # (1) check if all gradients are of the same type. Aggregating a mix of |
no test coverage detected