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.cc92 UnaryVariantOpRegistry::GetDeviceCopyFn( in GetDeviceCopyFn() function in tensorflow::UnaryVariantOpRegistry
102 AsyncVariantDeviceCopyFn* existing = GetDeviceCopyFn(direction, type_index); in RegisterDeviceCopyFn()
118 UnaryVariantOpRegistry::Global()->GetDeviceCopyFn(direction, in VariantDeviceCopy()
Dvariant_op_registry_test.cc139 EXPECT_EQ(UnaryVariantOpRegistry::Global()->GetDeviceCopyFn( in TEST()
144 auto* copy_to_gpu_fn = UnaryVariantOpRegistry::Global()->GetDeviceCopyFn( in TEST()
Dvariant_op_registry.h109 AsyncVariantDeviceCopyFn* GetDeviceCopyFn(