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) |
KernelAndDeviceOp(Rendezvous rendez,
boolean log_memory,
FunctionLibraryRuntime flr,
Pointer runner,
CollectiveExecutor.Handle collective_executor,
Device host_cpu_device) |
KernelAndDeviceOp(Rendezvous rendez,
boolean log_memory,
FunctionLibraryRuntime flr,
Pointer runner,
CollectiveExecutor.Handle collective_executor,
Device host_cpu_device) |
OpKernelConstruction(DeviceType device_type,
DeviceBase device,
Allocator allocator,
NodeDef node_def,
OpDef op_def,
FunctionLibraryRuntime flib,
DataTypeVector input_types,
StringAttrPairVector input_memory_types,
DataTypeVector output_types,
StringAttrPairVector output_memory_types,
int graph_def_version,
Status status) |
OpKernelConstruction(DeviceType device_type,
DeviceBase device,
Allocator allocator,
NodeDef node_def,
OpDef op_def,
FunctionLibraryRuntime flib,
DataTypeVector input_types,
StringAttrPairVector input_memory_types,
DataTypeVector output_types,
StringAttrPairVector output_memory_types,
int graph_def_version,
Status status) |