Searched refs:param_resolver (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 32 std::unique_ptr<ParamResolverInterface> param_resolver, in CollectiveExecutorMgr() argument
36 param_resolver_(std::move(param_resolver)), in CollectiveExecutorMgr()
101 auto param_resolver = std::make_unique<CollectiveParamResolverLocal>( in CreateProdLocalCollectiveExecutorMgr() local
105 config, device_mgr, std::move(device_resolver), std::move(param_resolver), in CreateProdLocalCollectiveExecutorMgr()
|
| D | test_collective_executor_mgr.h | 72 explicit TestCollectiveExecutorMgr(ParamResolverInterface* param_resolver, in TestCollectiveExecutorMgr() argument
74 : param_resolver_(param_resolver), rma_(rma) {} in TestCollectiveExecutorMgr()
|
| D | collective_executor_mgr.h | 31 std::unique_ptr<ParamResolverInterface> param_resolver,
|
| D | collective_test_util.h | 78 std::unique_ptr<ParamResolverInterface> param_resolver; member
|
| D | collective_test_util.cc | 145 test_env->param_resolver = std::make_unique<TestParamResolver>(); in CreateCollectiveTestEnv()
149 test_env->param_resolver.get(), /*rma=*/nullptr); in CreateCollectiveTestEnv()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr.cc | 31 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver, in RpcCollectiveExecutorMgr() argument
35 std::move(param_resolver), in RpcCollectiveExecutorMgr()
177 auto param_resolver = std::make_unique<CollectiveParamResolverDistributed>( in CreateProdRpcCollectiveExecutorMgr() local
181 config, device_mgr, std::move(dev_resolver), std::move(param_resolver), in CreateProdRpcCollectiveExecutorMgr()
|
| D | rpc_collective_executor_mgr.h | 40 std::unique_ptr<CollectiveParamResolverDistributed> param_resolver,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 216 CollectiveImplementationInterface** implementation, bool param_resolver) { in LookupHelper() argument
220 if (param_resolver) { in LookupHelper()
|
| D | collective.h | 492 bool param_resolver);
|