fix a bug

paddle/fluid/operators/lookup_table_op.cc

@@ -115,7 +115,7 @@ class LookupTableOpGrad : public framework::OperatorWithKernel {
  protected:
   framework::OpKernelType GetExpectedKernelType(
       const framework::ExecutionContext& ctx) const override {
-    auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("W"));
+    auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("Out"));
     return framework::OpKernelType(data_type, ctx.device_context());
   }
 };

python/paddle/fluid/transpiler/distribute_transpiler.py

@@ -487,7 +487,7 @@ class DistributeTranspiler(object):
             if init_op_num != 1:
                 raise ValueError("table init op num should be 1, now is " + str(
                     init_op_num))
-            table_init_op = table_param_init_op[1]
+            table_init_op = table_param_init_op[0]
             self.startup_program.global_block().append_op(
                 type="fake_init",
                 inputs={},