Searched refs:cpu_dispatch_key (Results 1 – 2 of 2) sorted by relevance
91 c10::DispatchKey cpu_dispatch_key) { in cpu_fallback() argument92 TORCH_CHECK(c10::BackendComponent::CPUBit == c10::toBackendComponent(cpu_dispatch_key), in cpu_fallback()94 c10::toString(cpu_dispatch_key)); in cpu_fallback()150 op.redispatchBoxed(c10::DispatchKeySet(cpu_dispatch_key), stack); in cpu_fallback()
15 c10::DispatchKey cpu_dispatch_key = c10::DispatchKey::CPU);