Searched refs:below_min (Results 1 – 3 of 3) sorted by relevance
179 const auto below_min = in operator() local182 backprop_wrt_min.device(d) = (gradients * below_min).sum(); in operator()262 const auto below_min = in operator() local267 (gradients_chip * below_min).sum(reduce); in operator()
6 NAME = 'policy_PolicyRefreshRate.below_min'
260 xla::XlaOp below_min = xla::Lt(input, nudged_input_min); in Compile() local261 xla::XlaOp select1 = xla::Select(below_min, gradient, zeroes); in Compile()