Searched refs:_push_weight_inputs (Results 1 – 1 of 1) sorted by relevance
108 self.push_weight_indices = self._push_weight_inputs(self.network.parameters_and_names())135 def _push_weight_inputs(weights): member in TrainOneStepCellWithServerCommunicator