Function InferAdadeltaUpdateTensorDesc

oneflow/user/ops/model_update_ops.cpp:317–327

}

Maybe<void> InferAdadeltaUpdateTensorDesc(user_op::InferContext* ctx) {
  const user_op::TensorDesc& model = ctx->InputTensorDesc("model", 0);
  const user_op::TensorDesc& model_diff = ctx->InputTensorDesc("model_diff", 0);
  const user_op::TensorDesc& square_avgs = ctx->InputTensorDesc("square_avgs", 0);
  const user_op::TensorDesc& acc_deltas = ctx->InputTensorDesc("acc_deltas", 0);
  JUST(CheckShapeLike(&model_diff, &model));
  JUST(CheckShapeLike(&square_avgs, &model));
  JUST(CheckShapeLike(&acc_deltas, &model));
  JUST(CheckLearningRateShape(ctx));
  return Maybe<void>::Ok();
}

Maybe<void> InferAdadeltaUpdateDataType(user_op::InferContext* ctx) {
  const user_op::TensorDesc& model = ctx->InputTensorDesc("model", 0);