CollectiveContext(CollectiveExecutor col_exec,
DeviceMgr dev_mgr,
OpKernelContext ctx,
OpKernelContext.Params op_params,
CollectiveParams col_params,
BytePointer exec_key,
long step_id,
Tensor input,
Tensor output) |
CollectiveContext(CollectiveExecutor col_exec,
DeviceMgr dev_mgr,
OpKernelContext ctx,
OpKernelContext.Params op_params,
CollectiveParams col_params,
BytePointer exec_key,
long step_id,
Tensor input,
Tensor output) |
CollectiveContext(CollectiveExecutor col_exec,
DeviceMgr dev_mgr,
OpKernelContext ctx,
OpKernelContext.Params op_params,
CollectiveParams col_params,
String exec_key,
long step_id,
Tensor input,
Tensor output) |
CollectiveContext(CollectiveExecutor col_exec,
DeviceMgr dev_mgr,
OpKernelContext ctx,
OpKernelContext.Params op_params,
CollectiveParams col_params,
String exec_key,
long step_id,
Tensor input,
Tensor output) |
Handle(CollectiveExecutor ce,
boolean inherit_ref) |
Handle(CollectiveExecutor ce,
boolean inherit_ref) |