Searched refs:compare_type (Results 1 – 2 of 2) sorted by relevance
125 PrimitiveType compare_type = operand_types[0]; in CreateScalarComparisonComputation() local129 if (compare_type == BF16) { in CreateScalarComparisonComputation()130 compare_type = F32; in CreateScalarComparisonComputation()134 int64 bit_width = primitive_util::BitWidth(compare_type); in CreateScalarComparisonComputation()
1661 PrimitiveType compare_type = operand_types[0]; in Sort() local1665 if (compare_type == BF16) { in Sort()1666 compare_type = F32; in Sort()1670 int64 bit_width = primitive_util::BitWidth(compare_type); in Sort()