Searched refs:device_copy_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.h | 109 const AsyncVariantDeviceCopyFn& device_copy_fn) { in RegisterDeviceCopyFn() argument
118 std::make_pair(direction, type_index), device_copy_fn)); in RegisterDeviceCopyFn()
392 const LocalVariantDeviceCopyFn& device_copy_fn) { in UnaryVariantDeviceCopyRegistration() argument
397 [type_index_name, device_copy_fn]( in UnaryVariantDeviceCopyRegistration()
410 return device_copy_fn(t, t_out, device_copy_tensor_fn); in UnaryVariantDeviceCopyRegistration()
530 device_copy_fn) \ argument
532 __COUNTER__, T, direction, TypeIndex::Make<T>(), device_copy_fn)
535 ctr, T, direction, type_index, device_copy_fn) \ argument
537 ctr, T, direction, type_index, device_copy_fn)
540 ctr, T, direction, type_index, device_copy_fn) \ argument
[all …]
|
| D | variant_op_registry.cc | 122 UnaryVariantOpRegistry::AsyncVariantDeviceCopyFn* device_copy_fn = in VariantDeviceCopy() local
125 if (device_copy_fn == nullptr) { in VariantDeviceCopy()
131 return (*device_copy_fn)(from, to, copy_fn); in VariantDeviceCopy()
|