File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -74,12 +74,12 @@ class LoDTensor : public Tensor {
74
74
LoD lod () const { return lod_; }
75
75
76
76
/*
77
- * Get a element from LoD.
77
+ * Get the start offset and end offset of an element from LoD.
78
78
*/
79
- size_t lod_element (size_t level, size_t elem) const {
79
+ std::pair< size_t , size_t > lod_element (size_t level, size_t elem) const {
80
80
PADDLE_ENFORCE_LT (level, NumLevels ());
81
81
PADDLE_ENFORCE_LT (elem, NumElements (level));
82
- return ( lod_)[level][elem];
82
+ return std::make_pair (( lod_)[level][elem], (lod_)[level][elem + 1 ]) ;
83
83
}
84
84
85
85
/*
Original file line number Diff line number Diff line change @@ -36,8 +36,8 @@ TEST(LoDTensor, LoDInGPU) {
36
36
lod_tensor.mutable_data <float >(place);
37
37
38
38
lod_tensor.set_lod (src_lod);
39
- CHECK_EQ (lod_tensor.lod_element (0 , 2 ), 4UL );
40
- CHECK_EQ (lod_tensor.lod_element (0 , 4 ), 8UL );
39
+ CHECK_EQ (lod_tensor.lod_element (0 , 2 ). first , 4UL );
40
+ CHECK_EQ (lod_tensor.lod_element (0 , 4 ). first , 8UL );
41
41
42
42
auto lod = lod_tensor.lod ();
43
43
You can’t perform that action at this time.
0 commit comments