Searched refs:safe_sub (Results 1 – 2 of 2) sorted by relevance
154 auto safe_sub = xla::Select(xla::Eq(labels, xla::ZerosLike(labels)), in CrossEntropyWithLogits() local156 auto mul = xla::Mul(xla::Neg(labels), safe_sub); in CrossEntropyWithLogits()
72 inline uint64 safe_sub(uint64 x, uint64 y) { return x >= y ? x - y : 0; } in safe_sub() function151 metrics::RecordTFDataGetNextDuration(safe_sub(end_time_us, start_time_us)); in GetNext()155 safe_sub(end_time_us, get_next_end_time_us_)); in GetNext()160 safe_sub(get_next_end_time_us_, get_next_start_time_us_)); in GetNext()