Searched refs:array_reduce (Results 1 – 2 of 2) sorted by relevance
193 return F.array_reduce(F.scalar_add, dout, F.shape(arr)), C.zeros_like(shp)
228 array_reduce = Primitive('array_reduce') variable