Searched refs:AsyncTensorDeviceCopyFn (Results 1 – 6 of 6) sorted by relevance
86 AsyncTensorDeviceCopyFn; typedef92 AsyncTensorDeviceCopyFn copy_fn)>254 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy_fn);341 UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn)>351 UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn in UnaryVariantDeviceCopyRegistration()
116 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy_fn) { in VariantDeviceCopy()133 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copier) { in DeviceCopyPrimitiveType()
169 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in WrappedDatasetVariantDeviceCopy()
28 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in OptionalDeviceCopy()
363 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in WrappedTensorDeviceCopy()
76 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in TensorListDeviceCopy()