Searched refs:reshaped_condition (Results 1 – 1 of 1) sorted by relevance
160 XlaOp reshaped_condition = xla::Reshape(condition, {flattened_size}); in CompileWhereWithSort() local161 XlaOp zeros = xla::ZerosLike(reshaped_condition); in CompileWhereWithSort()162 XlaOp compared = xla::Ne(reshaped_condition, zeros); in CompileWhereWithSort()209 XlaOp reshaped_condition = xla::Reshape(condition, {flattened_size}); in CompileWhereWithPrefixSum() local210 XlaOp zeros = xla::ZerosLike(reshaped_condition); in CompileWhereWithPrefixSum()212 xla::ConvertElementType(xla::Ne(reshaped_condition, zeros), S32); in CompileWhereWithPrefixSum()