Package | Description |
---|---|
org.bytedeco.tensorflow |
Modifier and Type | Method and Description |
---|---|
DeviceMgr |
CollectiveContext.dev_mgr() |
DeviceMgr |
CollectiveContext.dev_mgr() |
DeviceMgr |
FunctionLibraryRuntime.device_mgr() |
DeviceMgr |
ProcessFunctionLibraryRuntime.device_mgr() |
DeviceMgr |
WorkerEnv.device_mgr() |
DeviceMgr |
WorkerSession.device_mgr() |
DeviceMgr |
WorkerSession.device_mgr() |
DeviceMgr |
WorkerEnv.device_mgr() |
DeviceMgr |
ProcessFunctionLibraryRuntime.device_mgr() |
DeviceMgr |
FunctionLibraryRuntime.device_mgr() |
DeviceMgr |
EagerContext.local_device_mgr() |
DeviceMgr |
EagerContext.local_device_mgr() |
DeviceMgr |
DistributedFunctionLibraryRuntime.remote_device_mgr() |
DeviceMgr |
EagerContext.remote_device_mgr() |
DeviceMgr |
WorkerSession.remote_device_mgr() |
DeviceMgr |
WorkerSession.remote_device_mgr() |
DeviceMgr |
EagerContext.remote_device_mgr() |
DeviceMgr |
DistributedFunctionLibraryRuntime.remote_device_mgr() |
Modifier and Type | Method and Description |
---|---|
Status |
OpKernelContext.create_rendezvous(long step_id,
DeviceMgr device_mgr,
PointerPointer r) |
Status |
OpKernelContext.create_rendezvous(long step_id,
DeviceMgr device_mgr,
PointerPointer r) |
Status |
OpKernelContext.create_rendezvous(long step_id,
DeviceMgr device_mgr,
Rendezvous r) |
Status |
OpKernelContext.create_rendezvous(long step_id,
DeviceMgr device_mgr,
Rendezvous r) |
static WorkerSession |
WorkerSession.CreateWithBorrowedDeviceMgr(BytePointer session_name,
BytePointer worker_name,
WorkerCacheInterface worker_cache,
DeviceMgr borrowed_device_mgr,
GraphMgr graph_mgr,
DeviceMgr remote_device_mgr) |
static WorkerSession |
WorkerSession.CreateWithBorrowedDeviceMgr(BytePointer session_name,
BytePointer worker_name,
WorkerCacheInterface worker_cache,
DeviceMgr borrowed_device_mgr,
GraphMgr graph_mgr,
DeviceMgr remote_device_mgr) |
static WorkerSession |
WorkerSession.CreateWithBorrowedDeviceMgr(String session_name,
String worker_name,
WorkerCacheInterface worker_cache,
DeviceMgr borrowed_device_mgr,
GraphMgr graph_mgr,
DeviceMgr remote_device_mgr) |
static WorkerSession |
WorkerSession.CreateWithBorrowedDeviceMgr(String session_name,
String worker_name,
WorkerCacheInterface worker_cache,
DeviceMgr borrowed_device_mgr,
GraphMgr graph_mgr,
DeviceMgr remote_device_mgr) |
CollectiveContext |
CollectiveContext.dev_mgr(DeviceMgr setter) |
CollectiveContext |
CollectiveContext.dev_mgr(DeviceMgr setter) |
WorkerEnv |
WorkerEnv.device_mgr(DeviceMgr setter) |
WorkerEnv |
WorkerEnv.device_mgr(DeviceMgr setter) |
Status |
EagerContext.InitializeRemoteMaster(ServerInterface server,
WorkerEnv worker_env,
WorkerSession worker_session,
EagerClientCache remote_eager_workers,
DeviceMgr remote_device_manager,
StringVector remote_contexts,
long context_id,
Rendezvous r,
DeviceMgr local_device_mgr,
int keep_alive_secs,
DistributedFunctionLibraryRuntime cluster_flr,
RemoteMgr remote_mgr) |
Status |
EagerContext.InitializeRemoteMaster(ServerInterface server,
WorkerEnv worker_env,
WorkerSession worker_session,
EagerClientCache remote_eager_workers,
DeviceMgr remote_device_manager,
StringVector remote_contexts,
long context_id,
Rendezvous r,
DeviceMgr local_device_mgr,
int keep_alive_secs,
DistributedFunctionLibraryRuntime cluster_flr,
RemoteMgr remote_mgr) |
Status |
EagerContext.InitializeRemoteWorker(EagerClientCache remote_eager_workers,
DeviceMgr remote_device_mgr,
StringVector remote_contexts,
long context_id,
tensorflow.RendezvousCreator rendezvous_creator,
RemoteMgr remote_mgr) |
Status |
EagerContext.InitializeRemoteWorker(EagerClientCache remote_eager_workers,
DeviceMgr remote_device_mgr,
StringVector remote_contexts,
long context_id,
tensorflow.RendezvousCreator rendezvous_creator,
RemoteMgr remote_mgr) |
Status |
Session.LocalDeviceManager(DeviceMgr output) |
Status |
Session.LocalDeviceManager(DeviceMgr output) |
Status |
EagerContext.StoreCollectiveOpsServer(ServerInterface server,
DeviceMgr device_mgr,
CollectiveExecutorMgrInterface rpc_collective_executor_mgr) |
Status |
EagerContext.StoreCollectiveOpsServer(ServerInterface server,
DeviceMgr device_mgr,
CollectiveExecutorMgrInterface rpc_collective_executor_mgr) |
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) |
EagerContext(SessionOptions opts,
int default_device_placement_policy,
int default_mirroring_policy,
boolean async,
DeviceMgr device_mgr,
boolean device_mgr_owned,
Rendezvous rendezvous,
CustomKernelCreator custom_kernel_creator) |
EagerContext(SessionOptions opts,
int default_device_placement_policy,
int default_mirroring_policy,
boolean async,
DeviceMgr device_mgr,
boolean device_mgr_owned,
Rendezvous rendezvous,
CustomKernelCreator custom_kernel_creator) |
EagerContext(SessionOptions opts,
int default_device_placement_policy,
int default_mirroring_policy,
boolean async,
DeviceMgr device_mgr,
boolean device_mgr_owned,
Rendezvous rendezvous,
CustomKernelCreator custom_kernel_creator,
DistributedFunctionLibraryRuntime cluster_flr) |
EagerContext(SessionOptions opts,
int default_device_placement_policy,
int default_mirroring_policy,
boolean async,
DeviceMgr device_mgr,
boolean device_mgr_owned,
Rendezvous rendezvous,
CustomKernelCreator custom_kernel_creator,
DistributedFunctionLibraryRuntime cluster_flr) |
ProcessFunctionLibraryRuntime(DeviceMgr device_mgr,
Env env,
int graph_def_version,
FunctionLibraryDefinition lib_def,
OptimizerOptions optimizer_options) |
ProcessFunctionLibraryRuntime(DeviceMgr device_mgr,
Env env,
int graph_def_version,
FunctionLibraryDefinition lib_def,
OptimizerOptions optimizer_options) |
ProcessFunctionLibraryRuntime(DeviceMgr device_mgr,
Env env,
int graph_def_version,
FunctionLibraryDefinition lib_def,
OptimizerOptions optimizer_options,
ThreadPool thread_pool,
DistributedFunctionLibraryRuntime parent,
CustomKernelCreator custom_kernel_creator,
SessionMetadata metadata) |
ProcessFunctionLibraryRuntime(DeviceMgr device_mgr,
Env env,
int graph_def_version,
FunctionLibraryDefinition lib_def,
OptimizerOptions optimizer_options,
ThreadPool thread_pool,
DistributedFunctionLibraryRuntime parent,
CustomKernelCreator custom_kernel_creator,
SessionMetadata metadata) |
TFE_Context(SessionOptions opts,
int default_device_placement_policy,
Pointer default_mirroring_policy,
boolean async,
DeviceMgr device_mgr,
boolean device_mgr_owned,
Rendezvous rendezvous,
CustomKernelCreator custom_kernel_creator) |
TFE_Context(SessionOptions opts,
int default_device_placement_policy,
Pointer default_mirroring_policy,
boolean async,
DeviceMgr device_mgr,
boolean device_mgr_owned,
Rendezvous rendezvous,
CustomKernelCreator custom_kernel_creator) |
WorkerSession(BytePointer session_name,
BytePointer worker_name,
WorkerCacheInterface worker_cache,
DeviceMgr device_mgr,
GraphMgr graph_mgr,
DeviceMgr remote_device_mgr) |
WorkerSession(BytePointer session_name,
BytePointer worker_name,
WorkerCacheInterface worker_cache,
DeviceMgr device_mgr,
GraphMgr graph_mgr,
DeviceMgr remote_device_mgr) |
WorkerSession(String session_name,
String worker_name,
WorkerCacheInterface worker_cache,
DeviceMgr device_mgr,
GraphMgr graph_mgr,
DeviceMgr remote_device_mgr) |
WorkerSession(String session_name,
String worker_name,
WorkerCacheInterface worker_cache,
DeviceMgr device_mgr,
GraphMgr graph_mgr,
DeviceMgr remote_device_mgr) |
Copyright © 2022. All rights reserved.