KernelAndDeviceFunc(FunctionLibraryRuntime flr,
ProcessFunctionLibraryRuntime pflr,
DeviceVector input_devices,
IntTensorShapeMap input_tensor_shapes,
DtypeAndPartialTensorShapeIntMap input_resource_dtypes_and_shapes,
Pointer runner,
CollectiveExecutor.Handle collective_executor,
Device host_cpu_device,
BytePointer name,
tensorflow.RendezvousCreator rendezvous_creator) |
KernelAndDeviceFunc(FunctionLibraryRuntime flr,
ProcessFunctionLibraryRuntime pflr,
DeviceVector input_devices,
IntTensorShapeMap input_tensor_shapes,
DtypeAndPartialTensorShapeIntMap input_resource_dtypes_and_shapes,
Pointer runner,
CollectiveExecutor.Handle collective_executor,
Device host_cpu_device,
BytePointer name,
tensorflow.RendezvousCreator rendezvous_creator) |
KernelAndDeviceFunc(FunctionLibraryRuntime flr,
ProcessFunctionLibraryRuntime pflr,
DeviceVector input_devices,
IntTensorShapeMap input_tensor_shapes,
DtypeAndPartialTensorShapeIntMap input_resource_dtypes_and_shapes,
Pointer runner,
CollectiveExecutor.Handle collective_executor,
Device host_cpu_device,
String name,
tensorflow.RendezvousCreator rendezvous_creator) |
KernelAndDeviceFunc(FunctionLibraryRuntime flr,
ProcessFunctionLibraryRuntime pflr,
DeviceVector input_devices,
IntTensorShapeMap input_tensor_shapes,
DtypeAndPartialTensorShapeIntMap input_resource_dtypes_and_shapes,
Pointer runner,
CollectiveExecutor.Handle collective_executor,
Device host_cpu_device,
String name,
tensorflow.RendezvousCreator rendezvous_creator) |