Searched refs:between_nudged_min_max (Results 1 – 2 of 2) sorted by relevance
209 xla::XlaOp between_nudged_min_max = xla::And( in Compile() local213 xla::XlaOp output = xla::Select(between_nudged_min_max, gradient, zeroes); in Compile()304 xla::XlaOp between_nudged_min_max = xla::And( in Compile() local308 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()