Searched refs:above_max (Results 1 – 3 of 3) sorted by relevance
184 const auto above_max = in operator() local187 backprop_wrt_max.device(d) = (gradients * above_max).sum(); in operator()269 const auto above_max = in operator() local273 (gradients_chip * above_max).sum(reduce); in operator()
6 NAME = 'policy_PolicyRefreshRate.above_max'
269 xla::XlaOp above_max = xla::Gt(input, nudged_input_max); in Compile() local270 xla::XlaOp select2 = xla::Select(above_max, gradient, zeroes); in Compile()