Searched refs:RegisterDeviceCopyFn (Results 1 – 3 of 3) sorted by relevance
170 registry.RegisterDeviceCopyFn(VariantDeviceCopyDirection::HOST_TO_DEVICE, in TEST()172 EXPECT_DEATH(registry.RegisterDeviceCopyFn( in TEST()
103 void RegisterDeviceCopyFn(const VariantDeviceCopyDirection direction,347 UnaryVariantOpRegistry::Global()->RegisterDeviceCopyFn( in UnaryVariantDeviceCopyRegistration()
99 void UnaryVariantOpRegistry::RegisterDeviceCopyFn( in RegisterDeviceCopyFn() function in tensorflow::UnaryVariantOpRegistry