Searched refs:AsyncTensorDeviceCopyFn (Results 1 – 9 of 9) sorted by relevance
121 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy_fn) { in VariantDeviceCopy()138 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copier) { in DeviceCopyPrimitiveType()
90 AsyncTensorDeviceCopyFn; typedef96 AsyncTensorDeviceCopyFn copy_fn)>301 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy_fn);388 UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn)>399 UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn in UnaryVariantDeviceCopyRegistration()
192 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in WrappedDatasetVariantDeviceCopy()
57 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in RaggedTensorVariantDeviceCopy()
48 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in TensorMapDeviceCopy()
56 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in TensorListDeviceCopy()
31 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in OptionalDeviceCopy()
312 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in WrappedTensorDeviceCopy()
384 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in DeviceCopy()