Home
last modified time | relevance | path

Searched refs:input_u32_r2 (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtopk_ops.cc294 xla::XlaOp input_u32_r2 = xla::BitcastConvertType(input, xla::U32); in CreateMakeUnique() local
295 xla::XlaOp abs_r2 = xla::And(input_u32_r2, low_bit_mask_r2); in CreateMakeUnique()
298 xla::Or(input_u32_r2, smallest_normal_r2); in CreateMakeUnique()
300 xla::Select(if_zero_r2, smallest_normal_preserving_sign_r2, input_u32_r2); in CreateMakeUnique()