Searched refs:rebounded_result (Results 1 – 2 of 2) sorted by relevance
192 StatusOr<XlaOp> rebounded_result = xla::SetDimensionSizeWithRebound( in CompileWhereWithSort() local194 if (rebounded_result.ok()) { in CompileWhereWithSort()195 return rebounded_result; in CompileWhereWithSort()310 StatusOr<XlaOp> rebounded_result = xla::SetDimensionSizeWithRebound( in CompileWhereWithPrefixSum() local312 if (rebounded_result.ok()) { in CompileWhereWithPrefixSum()313 return *rebounded_result; in CompileWhereWithPrefixSum()
601 StatusOr<xla::XlaOp> rebounded_result = xla::SetDimensionSizeWithRebound( in ComputeResult() local603 if (rebounded_result.ok()) { in ComputeResult()604 selected_indices = *rebounded_result; in ComputeResult()