Searched refs:AscendLaunchTransData (Results 1 – 3 of 3) sorted by relevance
25 void AscendLaunchTransData::FreeDeviceMem(void *addr) { AscendLaunchKernel::FreeDeviceMem(addr); } in FreeDeviceMem()27 size_t AscendLaunchTransData::AlignSizeForLaunchKernel(size_t size) { in AlignSizeForLaunchKernel()31 uint8_t *AscendLaunchTransData::AllocDeviceMem(size_t size) { return AscendLaunchKernel::AllocDevic… in AllocDeviceMem()33 void AscendLaunchTransData::KernelSelect(const std::shared_ptr<session::KernelGraph> &kernel_graph)… in KernelSelect()37 void AscendLaunchTransData::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild()41 void AscendLaunchTransData::LaunchOpKernel() { in LaunchOpKernel()74 void AscendLaunchTransData::FreeLaunchDeviceMem() { in FreeLaunchDeviceMem()79 std::shared_ptr<session::KernelGraph> AscendLaunchTransData::ObtainTransDataKernelGraph() { in ObtainTransDataKernelGraph()93 void AscendLaunchTransData::ConstructKernelGraphAndSetAttr() { in ConstructKernelGraphAndSetAttr()
26 class AscendLaunchTransData : public AscendLaunchKernel {28 …AscendLaunchTransData(void *stream, TypeId dtype, size_t total_size, std::string src_format, std::… in AscendLaunchTransData() function39 ~AscendLaunchTransData() override = default;
266 …std::make_shared<AscendLaunchTransData>(stream, type_id_, size_, ori_format, dst_format, host_shap… in CreateLaunchTransData()