Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h115 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDevice() argument
120 collective_executor_(std::move(collective_executor)), in KernelAndDevice()
196 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDeviceOp() argument
198 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceOp()
267 std::unique_ptr<CollectiveExecutor::Handle> collective_executor, in KernelAndDeviceFunc() argument
272 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceFunc()
Dkernel_and_device.cc284 params.collective_executor = in Run()
385 opts->collective_executor = in RunAsync()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc61 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
106 c->collective_executor()->RunClosure([this, c, col_exec, done]() { in CanProceedWithCompute()
501 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
556 c->collective_executor()->RunClosure([c, in ComputeAsync()
630 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
693 c->collective_executor()->RunClosure([c, in ComputeAsync()
764 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
819 c->collective_executor()->RunClosure([c, in ComputeAsync()
890 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync()
948 c->collective_executor()->RunClosure([c, in ComputeAsync()
Dpartitioned_function_ops.cc246 run_opts.collective_executor = ctx->collective_executor(); in RunFunction()
Dfunction_ops.cc226 opts.collective_executor = ctx->collective_executor(); in ComputeAsync()
Dbatch_kernels.cc161 opts.collective_executor = last_task_context->collective_executor(); in ProcessFuncBatchImpl()
Dfunctional_ops.cc117 opts->collective_executor = ctx->collective_executor(); in SetRunOptions()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.h106 CollectiveExecutor* collective_executor = nullptr; member
Dgraph_runner.cc183 args.collective_executor = nullptr; in Run()
Ddirect_session.cc561 run_state.collective_executor.reset(new CollectiveExecutor::Handle( in RunInternal()
644 args.collective_executor = in RunInternal()
645 (run_state.collective_executor ? run_state.collective_executor->get() in RunInternal()
993 args.collective_executor = nullptr; in PRunSetup()
Dfunction.cc535 opts.collective_executor = ctx->collective_executor(); in ComputeAsync()
992 exec_args->collective_executor = run_opts.collective_executor; in ExecutorArgsFromOptions()
Ddirect_session.h202 std::unique_ptr<CollectiveExecutor::Handle> collective_executor; member
Dexecutor.cc398 collective_executor_(args.collective_executor), in ExecutorState()
699 params.collective_executor = collective_executor_; in Process()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h652 CollectiveExecutor* collective_executor = nullptr; member
1105 CollectiveExecutor* collective_executor() const { in collective_executor() function
1106 return params_->collective_executor; in collective_executor()
Dfunction.h791 CollectiveExecutor* collective_executor = nullptr; member
Dfunction.cc1762 " collective_executor=", IsSet(collective_executor), in DebugString()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc533 args.collective_executor = ce_handle ? ce_handle->get() : nullptr; in StartParallelExecutors()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc285 params->collective_executor = nullptr; in PrepOpKernel()