Searched refs:_make_input_context (Results 1 – 2 of 2) sorted by relevance
586 def _make_input_context(self): member in CollectiveAllReduceExtended601 input_context = self._make_input_context()617 input_context = self._make_input_context()637 input_context = self._make_input_context()650 input_context = self._make_input_context()
659 input_context=distribution.extended._make_input_context())720 input_context=distribution.extended._make_input_context())951 input_context=distribution.extended._make_input_context())1324 input_context=distribution.extended._make_input_context())1392 input_context=distribution.extended._make_input_context())1453 input_context=distribution.extended._make_input_context())