Searched defs:col_exec (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 74 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
112 bool CanProceedWithCompute(OpKernelContext* c, CollectiveExecutor* col_exec, in CanProceedWithCompute()
176 void ComputeAsyncImpl(OpKernelContext* c, CollectiveExecutor* col_exec, in ComputeAsyncImpl()
290 void ComputeAsyncImpl(OpKernelContext* c, CollectiveExecutor* col_exec, in ComputeAsyncImpl()
365 void ComputeAsyncImpl(OpKernelContext* c, CollectiveExecutor* col_exec, in ComputeAsyncImpl()
442 void ComputeAsyncImpl(OpKernelContext* c, CollectiveExecutor* col_exec, in ComputeAsyncImpl()
628 CollectiveExecutor* col_exec = c->collective_executor(); in Run() local
1009 auto* col_exec = c->collective_executor(); in ComputeAsync() local
1102 CollectiveExecutor* col_exec = c->collective_executor(); in Run() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_test_util.h | 83 core::RefCountPtr<CollectiveExecutor> col_exec; member
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 156 CollectiveExecutor* col_exec, NcclCommunicatorInterface* nccl_communicator, in CollectiveContext()
|
| D | collective.h | 392 CollectiveExecutor* col_exec; // Not owned member
|