Searched defs:device_input (Results 1 – 5 of 5) sorted by relevance
32 …eResource::Sink(const mindspore::MSTensor &host_input, std::shared_ptr<DeviceTensor> *device_input… in Sink()
55 …dResource::Sink(const mindspore::MSTensor &host_input, std::shared_ptr<DeviceTensor> *device_input… in Sink()
243 std::shared_ptr<mindspore::dataset::DeviceTensor> device_input; in operator ()() local340 std::shared_ptr<dataset::DeviceTensor> device_input; in operator ()() local
42 long device_input; member
167 std::shared_ptr<mindspore::dataset::DeviceTensor> &device_input) { in H2D_Sink()