Searched refs:AsyncTensorDeviceCopyFn (Results 1 – 9 of 9) sorted by relevance
93 AsyncTensorDeviceCopyFn; typedef99 AsyncTensorDeviceCopyFn copy_fn)>302 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy_fn);389 UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn)>400 UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn in UnaryVariantDeviceCopyRegistration()
131 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy_fn) { in VariantDeviceCopy()148 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copier) { in DeviceCopyPrimitiveType()
209 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in WrappedDatasetVariantDeviceCopy()
57 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in RaggedTensorVariantDeviceCopy()
48 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in TensorMapDeviceCopy()
685 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in TensorListDeviceCopy()
31 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in OptionalDeviceCopy()
320 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in WrappedTensorDeviceCopy()
384 const UnaryVariantOpRegistry::AsyncTensorDeviceCopyFn& copy) { in DeviceCopy()