Searched defs:FusedKernelCPU (Results 1 – 2 of 2) sorted by relevance
22 struct TORCH_API FusedKernelCPU : public FusedKernel { struct32 at::Backend backend() const override { in backend()36 void launch_raw(const uint32_t numel, std::vector<void*>& arguments) in launch_raw()43 void (*kernel)(uint32_t, void**) = nullptr;
306 FusedKernelCPU::FusedKernelCPU( in FusedKernelCPU() function in torch::jit::fuser::cpu::FusedKernelCPU