Searched defs:recv_at_host (Results 1 – 3 of 3) sorted by relevance
97 auto recv_at_host = rewriter.create<TF::_XlaRecvAtHostOp>( in matchAndRewrite() local
443 Operation* recv_at_host, Operation* insertion_point, in ReplaceExternalOperandUsage()548 Operation* recv_at_host = CreateRecvAtHostOp( in MoveOpsToHost() local
75 Node *recv_at_host = node_name_image["outside_compilation_cluster__0_recv"]; in TEST() local142 Node *recv_at_host = node_name_image["outside_compilation_cluster__0_recv"]; in TEST() local168 Node *recv_at_host = node_name_image["outside_compilation_cluster__0_recv"]; in TEST() local192 Node *recv_at_host = node_name_image["outside_compilation_cluster__0_recv"]; in TEST() local