Searched refs:DatasetIteratorKernel (Results 1 – 2 of 2) sorted by relevance
39 DatasetIteratorKernel::DatasetIteratorKernel() in DatasetIteratorKernel() function in mindspore::kernel::DatasetIteratorKernel42 DatasetIteratorKernel::~DatasetIteratorKernel() { GpuBufferMgr::GetInstance().Close(handle_); } in ~DatasetIteratorKernel()44 void DatasetIteratorKernel::ReleaseResource() { in ReleaseResource()49 const std::vector<size_t> &DatasetIteratorKernel::GetInputSizeList() const { return input_size_list… in GetInputSizeList()51 const std::vector<size_t> &DatasetIteratorKernel::GetOutputSizeList() const { return output_size_li… in GetOutputSizeList()53 const std::vector<size_t> &DatasetIteratorKernel::GetWorkspaceSizeList() const { return workspace_s… in GetWorkspaceSizeList()55 bool DatasetIteratorKernel::Init(const CNodePtr &kernel_node) { in Init()92 void DatasetIteratorKernel::InitSizeLists() { return; } in InitSizeLists()94 bool DatasetIteratorKernel::ReadDevice(void **addr, size_t *len) { in ReadDevice()136 bool DatasetIteratorKernel::Launch(const std::vector<AddressPtr> &, const std::vector<AddressPtr> &, in Launch()
29 class DatasetIteratorKernel : public GpuKernel {31 DatasetIteratorKernel();32 ~DatasetIteratorKernel();58 MS_REG_GPU_KERNEL(GetNext, DatasetIteratorKernel)