Searched refs:LaunchMul (Results 1 – 4 of 4) sorted by relevance
23 std::shared_ptr<session::KernelGraph> LaunchMul::ObtainMulKernelGraph() { in ObtainMulKernelGraph()40 kernel::KernelMod *LaunchMul::ObtainLaunchMulKernelMod() { in ObtainLaunchMulKernelMod()58 void LaunchMul::ObtainMulInputsAddr() { in ObtainMulInputsAddr()76 void LaunchMul::FreeInputDeviceMemory() { in FreeInputDeviceMemory()
25 class LaunchMul {27 LaunchMul(TypeId dtype, size_t total_size) in LaunchMul() function34 virtual ~LaunchMul() = default;
26 class GPULaunchMul : public GPULaunchkernel, public LaunchMul {28 …GPULaunchMul(void *stream, TypeId dtype, size_t total_size) : GPULaunchkernel(stream), LaunchMul(d… in GPULaunchMul()
26 class AscendLaunchMul : public AscendLaunchKernel, public LaunchMul {29 : AscendLaunchKernel(stream), LaunchMul(dtype, total_size) {} in AscendLaunchMul()