Home
last modified time | relevance | path

Searched refs:GetDeviceCopyFn (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry_test.cc156 EXPECT_EQ(UnaryVariantOpRegistry::Global()->GetDeviceCopyFn( in TEST()
161 auto* copy_to_gpu_fn = UnaryVariantOpRegistry::Global()->GetDeviceCopyFn( in TEST()
Dvariant_op_registry.cc123 UnaryVariantOpRegistry::Global()->GetDeviceCopyFn(direction, in VariantDeviceCopy()
Dvariant_op_registry.h110 AsyncVariantDeviceCopyFn* existing = GetDeviceCopyFn(direction, type_index); in RegisterDeviceCopyFn()
123 AsyncVariantDeviceCopyFn* GetDeviceCopyFn( in GetDeviceCopyFn() function