Searched defs:reshaped_condition (Results 1 – 1 of 1) sorted by relevance
160 XlaOp reshaped_condition = xla::Reshape(condition, {flattened_size}); in CompileWhereWithSort() local209 XlaOp reshaped_condition = xla::Reshape(condition, {flattened_size}); in CompileWhereWithPrefixSum() local