Searched refs:between_nudged_min_max (Results 1 – 2 of 2) sorted by relevance
167 xla::XlaOp between_nudged_min_max = xla::And( in Compile() local171 xla::XlaOp output = xla::Select(between_nudged_min_max, gradient, zeroes); in Compile()253 xla::XlaOp between_nudged_min_max = xla::And( in Compile() local257 xla::XlaOp output0 = xla::Select(between_nudged_min_max, gradient, zeroes); in Compile()
116 auto between_nudged_min_max = in operator() local119 backprops.device(d) = gradients * between_nudged_min_max; in operator()