Searched refs:stackBasedKernel (Results 1 – 1 of 1) sorted by relevance
305 void stackBasedKernel(const OperatorHandle&, c10::Stack* stack) { in stackBasedKernel() function312 .kernel<&stackBasedKernel>(c10::DispatchKey::CPU) in TEST()313 .kernel<&stackBasedKernel>(c10::DispatchKey::CUDA) in TEST()314 .kernel<&stackBasedKernel>(c10::DispatchKey::XLA) in TEST()315 .kernel<&stackBasedKernel>(c10::DispatchKey::Lazy)); in TEST()322 .kernel<&stackBasedKernel>(c10::DispatchKey::CPU) in TEST()323 .kernel<&stackBasedKernel>(c10::DispatchKey::CUDA) in TEST()324 .kernel<&stackBasedKernel>(c10::DispatchKey::XLA) in TEST()325 .kernel<&stackBasedKernel>(c10::DispatchKey::Lazy)); in TEST()351 .kernel<&stackBasedKernel>(c10::DispatchKey::CPU) in TEST()[all …]