Ejemplo n.º 1
0
 def fn(x):
   return xla.reduce(
       x, init_value=1, dimensions_to_reduce=dims, reducer=mul_reducer)
Ejemplo n.º 2
0
 def fn(x):
   return xla.reduce(
       x, init_value=1, dimensions_to_reduce=dims, reducer=mul_reducer)
Ejemplo n.º 3
0
 def fn(x):
   return xla.reduce(
       x, init_value=0, dimensions_to_reduce=dims, reducer=sum_reducer)
Ejemplo n.º 4
0
 def fn(x):
   return xla.reduce(
       x, init_value=0, dimensions_to_reduce=dims, reducer=sum_reducer)