Function KvApplyGradientDescentShapeFn

tensorflow/core/ops/training_ali_ops.cc:663–672

#undef REGISTER_OP_BY_NAME

static Status KvApplyGradientDescentShapeFn(InferenceContext* c) {
  ShapeHandle unused;
  TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 0, &unused));  // alpha
  ShapeHandle grad = ShapeOrHandleShape(c, 2);
  ShapeHandle indices;
  TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 1, &indices));
  DimensionHandle unused2;
  TF_RETURN_IF_ERROR(c->Merge(c->Dim(indices, 0), c->Dim(grad, 0), &unused2));
  return Status::OK();
}

#define REGISTER_OP_BY_NAME(name)              \
REGISTER_OP(name)                              \