Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc91 UnaryVariantOpRegistry::AsyncVariantDeviceCopyFn*
101 const AsyncVariantDeviceCopyFn& device_copy_fn) { in RegisterDeviceCopyFn()
102 AsyncVariantDeviceCopyFn* existing = GetDeviceCopyFn(direction, type_index); in RegisterDeviceCopyFn()
109 AsyncVariantDeviceCopyFn>(std::make_pair(direction, type_index), in RegisterDeviceCopyFn()
117 UnaryVariantOpRegistry::AsyncVariantDeviceCopyFn* device_copy_fn = in VariantDeviceCopy()
Dvariant_op_registry.h93 AsyncVariantDeviceCopyFn; typedef
105 const AsyncVariantDeviceCopyFn& device_copy_fn);
109 AsyncVariantDeviceCopyFn* GetDeviceCopyFn(
163 AsyncVariantDeviceCopyFn, PairHash>
Dvariant_op_registry_test.cc167 UnaryVariantOpRegistry::AsyncVariantDeviceCopyFn f; in TEST()