Skip to content

Commit 0d9d25d

Browse files
authored
Feature/refactor layers to Layers (#16337)
* test=develop, add some Layers and tests * test=develop, add more layers * test=develop, add more layers * test=develop, add force cpu option * Update test_layers.py remove pdb * test=develop, refine code
1 parent 850b737 commit 0d9d25d

File tree

3 files changed

+983
-11
lines changed

3 files changed

+983
-11
lines changed

paddle/fluid/operators/lookup_table_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class LookupTableOp : public framework::OperatorWithKernel {
3333
auto table_dims = ctx->GetInputDim("W");
3434
auto ids_dims = ctx->GetInputDim("Ids");
3535
int ids_rank = ids_dims.size();
36-
36+
VLOG(5) << "ids rank is " << ids_rank << std::endl;
3737
PADDLE_ENFORCE_EQ(table_dims.size(), 2);
3838
PADDLE_ENFORCE_EQ(ids_dims[ids_rank - 1], 1,
3939
"The last dimension of the 'Ids' tensor must be 1.");

0 commit comments

Comments
 (0)