|
|
@ -359,6 +359,9 @@ All parameter, weight, gradient are variables in Paddle.
|
|
|
|
return self.GetMutable<platform::Communicator>();
|
|
|
|
return self.GetMutable<platform::Communicator>();
|
|
|
|
},
|
|
|
|
},
|
|
|
|
py::return_value_policy::reference)
|
|
|
|
py::return_value_policy::reference)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifndef _WIN32
|
|
|
|
.def("get_reader",
|
|
|
|
.def("get_reader",
|
|
|
|
[](Variable &self) -> framework::ReaderHolder * {
|
|
|
|
[](Variable &self) -> framework::ReaderHolder * {
|
|
|
|
PADDLE_ENFORCE(self.IsType<framework::ReaderHolder>());
|
|
|
|
PADDLE_ENFORCE(self.IsType<framework::ReaderHolder>());
|
|
|
@ -366,7 +369,7 @@ All parameter, weight, gradient are variables in Paddle.
|
|
|
|
},
|
|
|
|
},
|
|
|
|
py::return_value_policy::reference)
|
|
|
|
py::return_value_policy::reference)
|
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
;
|
|
|
|
; // NOLINT
|
|
|
|
|
|
|
|
|
|
|
|
#if !defined(_WIN32)
|
|
|
|
#if !defined(_WIN32)
|
|
|
|
py::class_<framework::ReaderHolder>(m, "Reader", "")
|
|
|
|
py::class_<framework::ReaderHolder>(m, "Reader", "")
|
|
|
|