Searched refs:GetDeviceCopyFn (Results 1 – 3 of 3) sorted by relevance
92 UnaryVariantOpRegistry::GetDeviceCopyFn( in GetDeviceCopyFn() function in tensorflow::UnaryVariantOpRegistry102 AsyncVariantDeviceCopyFn* existing = GetDeviceCopyFn(direction, type_index); in RegisterDeviceCopyFn()118 UnaryVariantOpRegistry::Global()->GetDeviceCopyFn(direction, in VariantDeviceCopy()
139 EXPECT_EQ(UnaryVariantOpRegistry::Global()->GetDeviceCopyFn( in TEST()144 auto* copy_to_gpu_fn = UnaryVariantOpRegistry::Global()->GetDeviceCopyFn( in TEST()
109 AsyncVariantDeviceCopyFn* GetDeviceCopyFn(