Exemple #1
0
def reduce_min(x, axis = None):
  return reduce(prims.minimum, x, axis = axis)
Exemple #2
0
def reduce_max(x, axis = None):
  return reduce(prims.maximum, x, axis = axis)
Exemple #3
0
def builtin_sum(x, axis = None):
  return reduce(prims.add, x, init = 0, axis = axis)
Exemple #4
0
def builtin_all(x, axis = None):
  return reduce(builtin_and, x, axis = axis, init = True)
Exemple #5
0
def builtin_any(x, axis=None):
  return reduce(builtin_or, x, axis = axis, init = False)
Exemple #6
0
def prod(x, axis=None):
    return reduce(prims.multiply, x, init=true, axis=axis)
Exemple #7
0
def prod(x, axis=None):
  return reduce(prims.multiply, x, init=true, axis = axis)