Home
last modified time | relevance | path

Searched refs:collective_executor (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h107 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDevice() argument
112 collective_executor_(std::move(collective_executor)), in KernelAndDevice()
174 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDeviceOp() argument
176 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceOp()
237 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDeviceFunc() argument
241 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceFunc()
Dkernel_and_device.cc276 params.collective_executor = in Run()
347 opts->collective_executor = in Run()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc45 c->collective_executor()->RunClosure([this, c, done, col_exec]() { in CanProceedWithCompute()
94 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
218 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
290 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
366 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
Dpartitioned_function_ops.cc240 run_opts.collective_executor = ctx->collective_executor(); in RunFunction()
Dfunction_ops.cc258 opts.collective_executor = ctx->collective_executor(); in ComputeAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.h100 CollectiveExecutor* collective_executor = nullptr; member
Dgraph_runner.cc185 args.collective_executor = nullptr; in Run()
Ddirect_session.cc551 run_state.collective_executor.reset(new CollectiveExecutor::Handle( in RunInternal()
632 args.collective_executor = in RunInternal()
633 (run_state.collective_executor ? run_state.collective_executor->get() in RunInternal()
979 args.collective_executor = nullptr; in PRunSetup()
Ddirect_session.h202 std::unique_ptr<CollectiveExecutor::Handle> collective_executor; member
Dfunction.cc529 opts.collective_executor = ctx->collective_executor(); in ComputeAsync()
1011 exec_args->collective_executor = run_opts.collective_executor; in ExecutorArgsFromOptions()
Dexecutor.cc1455 collective_executor_(args.collective_executor), in ExecutorState()
1715 params.collective_executor = collective_executor_; in Process()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h685 CollectiveExecutor* collective_executor = nullptr; member
1115 CollectiveExecutor* collective_executor() const { in collective_executor() function
1116 return params_->collective_executor; in collective_executor()
Dfunction.h692 CollectiveExecutor* collective_executor = nullptr; member
Dfunction.cc1720 " collective_executor=", IsSet(collective_executor), in DebugString()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc519 args.collective_executor = ce_handle ? ce_handle->get() : nullptr; in StartParallelExecutors()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr_test.cc431 params->collective_executor = nullptr; in PrepOpKernel()