Home
last modified time | relevance | path

Searched defs:col_exec (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc61 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
99 bool CanProceedWithCompute(OpKernelContext* c, CollectiveExecutor* col_exec, in CanProceedWithCompute()
164 void ComputeAsyncImpl(OpKernelContext* c, CollectiveExecutor* col_exec, in ComputeAsyncImpl()
274 void ComputeAsyncImpl(OpKernelContext* c, CollectiveExecutor* col_exec, in ComputeAsyncImpl()
349 void ComputeAsyncImpl(OpKernelContext* c, CollectiveExecutor* col_exec, in ComputeAsyncImpl()
426 void ComputeAsyncImpl(OpKernelContext* c, CollectiveExecutor* col_exec, in ComputeAsyncImpl()
501 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
630 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
764 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
890 CollectiveExecutor* col_exec = c->collective_executor(); in ComputeAsync() local
/external/tensorflow/tensorflow/core/framework/
Dcollective.cc170 CollectiveExecutor* col_exec, NcclCommunicatorInterface* nccl_communicator, in CollectiveContext()
Dcollective.h365 CollectiveExecutor* col_exec; // Not owned member