Package | Description |
---|---|
org.bytedeco.tensorflow |
Modifier and Type | Method and Description |
---|---|
void |
CollectiveExecutor.CompleteParamsAsync(BytePointer device,
CollectiveParams cp,
CancellationManager cancel_mgr,
Pointer done) |
void |
ParamResolverInterface.CompleteParamsAsync(BytePointer device,
CollectiveParams cp,
CancellationManager cancel_mgr,
Pointer done) |
void |
ParamResolverInterface.CompleteParamsAsync(BytePointer device,
CollectiveParams cp,
CancellationManager cancel_mgr,
Pointer done) |
void |
CollectiveExecutor.CompleteParamsAsync(BytePointer device,
CollectiveParams cp,
CancellationManager cancel_mgr,
Pointer done) |
void |
CollectiveExecutor.CompleteParamsAsync(String device,
CollectiveParams cp,
CancellationManager cancel_mgr,
Pointer done) |
void |
ParamResolverInterface.CompleteParamsAsync(String device,
CollectiveParams cp,
CancellationManager cancel_mgr,
Pointer done) |
void |
ParamResolverInterface.CompleteParamsAsync(String device,
CollectiveParams cp,
CancellationManager cancel_mgr,
Pointer done) |
void |
CollectiveExecutor.CompleteParamsAsync(String device,
CollectiveParams cp,
CancellationManager cancel_mgr,
Pointer done) |
void |
CollectiveExecutor.ExecuteAsync(OpKernelContext ctx,
CollectiveParams col_params,
BytePointer exec_key,
Pointer done) |
void |
CollectiveExecutor.ExecuteAsync(OpKernelContext ctx,
CollectiveParams col_params,
BytePointer exec_key,
Pointer done) |
void |
CollectiveExecutor.ExecuteAsync(OpKernelContext ctx,
CollectiveParams col_params,
String exec_key,
Pointer done) |
void |
CollectiveExecutor.ExecuteAsync(OpKernelContext ctx,
CollectiveParams col_params,
String exec_key,
Pointer done) |
Status |
CollectiveImplementationInterface.InitializeCollectiveParams(CollectiveParams col_params) |
Status |
CollectiveImplementationInterface.InitializeCollectiveParams(CollectiveParams col_params) |
void |
CollectiveExecutor.Launched(CollectiveParams col_params) |
void |
CollectiveExecutor.Launched(CollectiveParams col_params) |
void |
CollectiveExecutor.WaitForDependencies(CollectiveParams col_params) |
void |
CollectiveExecutor.WaitForDependencies(CollectiveParams col_params) |
Constructor and Description |
---|
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) |
Copyright © 2022. All rights reserved.