• F
    Add Tensor::CopyFrom and Tensor::mutable_data(Place place) · 69d99d48
    fengjiayi 提交于
    1. Add `Tensor::CopyFrom`. Current version can only support CPU memory
    copy. The support of GPU will be provided later by `paddle::memory`.
    The current implementation of `Tensor::CopyFrom` is a little inefficient:
    Every time `CopyFrom` is called, tensor will re-allocate its memory. However, if
    we try to check and reuse `placeholder_`, we have to provide a template
    parameter for `CopyFrom` to indicate the data type. It seems strange for
    a simple copy function.
    
    2. Add `Tensor::mutable_data(Place place)`, which directly use member
    variable `dims_` as its dim parameter. This interface is required by
    `Op::InferShape`.
    69d99d48
tensor.h 5.7 KB