Searched refs:tpu_ordinal_function (Results 1 – 1 of 1) sorted by relevance
574 tpu_ordinal_function=None, argument621 if tpu_ordinal_function is None:622 tpu_ordinal_function = lambda index: -1 function629 tpu_ordinal=tpu_ordinal_function(index),649 tpu_ordinal_function=None): argument702 if tpu_ordinal_function is None:703 tpu_ordinal_function = self._default_ordinal_function714 if tpu_ordinal_function is None:715 tpu_ordinal_function = _ordinal_function_from_map749 tpu_ordinal=tpu_ordinal_function(index))