Searched refs:GetCollectiveExecutorHandle (Results 1 – 4 of 4) sorted by relevance
519 auto collective_executor_handle = context->GetCollectiveExecutorHandle(); in TFE_AbortCollectiveOps()528 auto collective_executor_handle = context->GetCollectiveExecutorHandle(); in TFE_CollectiveOpsCheckPeerHealth()
57 runner, ctx.GetCollectiveExecutorHandle(), in CreateUncachedKernelAndDeviceOp()
296 std::unique_ptr<CollectiveExecutor::Handle> GetCollectiveExecutorHandle() { in GetCollectiveExecutorHandle() function
531 ctx.GetCollectiveExecutorHandle(), ctx.HostCPU(), op->Name(), in GetOrCreateKernelAndDevice()538 ctx.GetCollectiveExecutorHandle(), ctx.HostCPU())); in GetOrCreateKernelAndDevice()