Searched refs:gradients_chip (Results 1 – 1 of 1) sorted by relevance
251 const auto gradients_chip = gradients.chip<1>(i); in operator() local255 backprops_wrt_input.chip<1>(i).device(d) = gradients_chip; in operator()270 gradients_chip * between_min_max; in operator()277 (gradients_chip * below_min).sum(reduce); in operator()283 (gradients_chip * above_max).sum(reduce); in operator()