Searched refs:device_copy_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.h | 105 const AsyncVariantDeviceCopyFn& device_copy_fn);
345 const LocalVariantDeviceCopyFn& device_copy_fn) { in UnaryVariantDeviceCopyRegistration() argument
349 [type_index_name, device_copy_fn]( in UnaryVariantDeviceCopyRegistration()
362 return device_copy_fn(t, t_out, device_copy_tensor_fn); in UnaryVariantDeviceCopyRegistration()
479 device_copy_fn) \ argument
481 __COUNTER__, T, direction, MakeTypeIndex<T>(), device_copy_fn)
484 ctr, T, direction, type_index, device_copy_fn) \ argument
486 ctr, T, direction, type_index, device_copy_fn)
489 ctr, T, direction, type_index, device_copy_fn) \ argument
493 direction, type_index, device_copy_fn)
|
| D | variant_op_registry.cc | 101 const AsyncVariantDeviceCopyFn& device_copy_fn) { in RegisterDeviceCopyFn() argument
110 device_copy_fn)); in RegisterDeviceCopyFn()
117 UnaryVariantOpRegistry::AsyncVariantDeviceCopyFn* device_copy_fn = in VariantDeviceCopy() local
120 if (device_copy_fn == nullptr) { in VariantDeviceCopy()
126 return (*device_copy_fn)(from, to, copy_fn); in VariantDeviceCopy()
|