Function GetIsPositive

tensorflow/cc/gradients/math_grad.cc:1262–1275

Helper function for unsorted segment ops. Returns a mask of where 'ids' are positive, reshaped so that it will be broadcastable to the result shape of gathering params by ids.

// Returns a mask of where 'ids' are positive, reshaped so that it will be
// broadcastable to the result shape of gathering params by ids.
Output GetIsPositive(const Scope& scope, const Output& params,
                     const Output& ids) {
  Output is_positive = GreaterEqual(scope, ids, ZerosLike(scope, ids));
  // tf.where(condition, x, y) requires condition to have the same shape as x
  // and y.
  Output is_positive_shape = Shape(scope, is_positive);
  Output ones =
      Tile(scope, Const(scope, {1}), Subtract(scope, Rank(scope, params), {1}));
  auto broadcastable_shape = Concat(scope, {is_positive_shape, ones},
                                    /*axis=*/0);
  is_positive = Reshape(scope, is_positive, broadcastable_shape);
  is_positive = LogicalAnd(scope, is_positive, OnesLike(scope, is_positive));
  return is_positive;
}

// Helper function for unsorted segment ops.
// Gathers params for positive segment ids and gathers 0 for inputs with