Using LODTensor instead of Tensor in every operator.
Created by: reyoung
Since we try to use LODTensor
instead of Tensor
as our basic data type for Input/Output variables of operators, we should:
-
Using
LODTensor
instead ofTensor
in current code. For example, in current implementation, like here, we should useInput<LODTensor>()
instead ofInput<Tensor>()
. -
How does output inherit
LOD
information from one of the operator's inputs?- Should that be registered in
OpInfo
. e.g. RegisterOutput["out"] inherits input["X"]'s lod information
? - Or is that should be hand written in operator implementation by each developer. (which could be very noising).
- Should that be registered in