Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand


tf.reduce_prod(t, axis) with faster custom gradient.

Shows comparable speed on CPU, up to 2x speed up on GPU, and 7x on TPU.

t The tensor to reduce.
axis The dimension to reduce.

prod(t) and grad(prod(t))