Searched refs:reduction_str (Results 1 – 2 of 2) sorted by relevance
28 def get_reduce_op(reduction_str): argument30 if reduction_str == "max":32 elif reduction_str == "mean":34 elif reduction_str == "min":36 elif reduction_str == "prod":38 elif reduction_str == "sum":42 reduction_str)
57 def test_unweighted_ragged_reduction(self, reduction_str, expected_output): argument62 output_tensor = reduction.Reduction(reduction=reduction_str)(input_tensor)95 def test_weighted_ragged_reduction(self, reduction_str, expected_output): argument104 output_tensor = reduction.Reduction(reduction=reduction_str)(149 def test_unweighted_dense_reduction(self, reduction_str, expected_output): argument154 output_tensor = reduction.Reduction(reduction=reduction_str)(input_tensor)187 def test_weighted_dense_reduction(self, reduction_str, expected_output): argument196 output_tensor = reduction.Reduction(reduction=reduction_str)(