Searched refs:GetDistributedContext (Results 1 – 3 of 3) sorted by relevance
179 mlir::Value CoreRTConverter::GetDistributedContext( in GetDistributedContext() function in tensorflow::CoreRTConverter208 mlir::Value dist_ctx = GetDistributedContext(op, rewriter); in GetRemoteChainManager()264 mlir::Value distributed_context = GetDistributedContext(op, rewriter); in GetTaskHandle()
82 mlir::Value GetDistributedContext(mlir::Operation *op,
490 corert_converter_.GetDistributedContext(op.getOperation(), &rewriter); in matchAndRewrite()