Searched refs:GetDeviceCopyFn (Results 1 – 3 of 3) sorted by relevance
156 EXPECT_EQ(UnaryVariantOpRegistry::Global()->GetDeviceCopyFn( in TEST()161 auto* copy_to_gpu_fn = UnaryVariantOpRegistry::Global()->GetDeviceCopyFn( in TEST()
123 UnaryVariantOpRegistry::Global()->GetDeviceCopyFn(direction, in VariantDeviceCopy()
110 AsyncVariantDeviceCopyFn* existing = GetDeviceCopyFn(direction, type_index); in RegisterDeviceCopyFn()123 AsyncVariantDeviceCopyFn* GetDeviceCopyFn( in GetDeviceCopyFn() function