Searched refs:above_max (Results 1 – 2 of 2) sorted by relevance
191 const auto above_max = in operator() local194 backprop_wrt_max.device(d) = (gradients * above_max).sum(); in operator()279 const auto above_max = in operator() local283 (gradients_chip * above_max).sum(reduce); in operator()
224 xla::XlaOp above_max = xla::Gt(input, nudged_input_max); in Compile() local225 xla::XlaOp select2 = xla::Select(above_max, gradient, zeroes); in Compile()354 xla::XlaOp above_max = xla::Gt(input, nudged_input_max); in Compile() local355 xla::XlaOp select2 = xla::Select(above_max, gradient, zeroes); in Compile()