Searched defs:DeviceShapeTransfer (Results 1 – 1 of 1) sorted by relevance
973 using DeviceShapeTransfer = std::function<std::vector<size_t>(const std::vector<size_t> &)>; in TransShapeToDevice() typedef1019 using DeviceShapeTransfer = std::function<std::vector<int64_t>(const std::vector<int64_t> &)>; in TransShapeToDevice() typedef