Home
last modified time | relevance | path

Searched defs:device_copy_fn (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.h345 const LocalVariantDeviceCopyFn& device_copy_fn) { in UnaryVariantDeviceCopyRegistration()
479 device_copy_fn) \ argument
484 ctr, T, direction, type_index, device_copy_fn) \ argument
489 ctr, T, direction, type_index, device_copy_fn) \ argument
Dvariant_op_registry.cc101 const AsyncVariantDeviceCopyFn& device_copy_fn) { in RegisterDeviceCopyFn()
117 UnaryVariantOpRegistry::AsyncVariantDeviceCopyFn* device_copy_fn = in VariantDeviceCopy() local