Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcollective_ops.cc74 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_test_util.h83 core::RefCountPtr<CollectiveExecutor> col_exec; member
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcollective.cc156 CollectiveExecutor* col_exec, NcclCommunicatorInterface* nccl_communicator, in CollectiveContext()
H A Dcollective.h392 CollectiveExecutor* col_exec; // Not owned member