Searched refs:gradients_chip (Results 1 – 1 of 1) sorted by relevance
241 const auto gradients_chip = gradients.chip<1>(i); in operator() local245 backprops_wrt_input.chip<1>(i).device(d) = gradients_chip; in operator()260 gradients_chip * between_min_max; in operator()267 (gradients_chip * below_min).sum(reduce); in operator()273 (gradients_chip * above_max).sum(reduce); in operator()