Searched defs:send_func (Results 1 – 4 of 4) sorted by relevance
118 auto send_func = send_cluster->getParentOfType<mlir::func::FuncOp>(); in LowerDTensorSendToXlaOp() local
262 TFE_Op* send_func = TFE_NewOp(ctx, "SendFunction", status); in TEST_P() local
241 def _test_send(self, send_func): argument
53 def __init__(self, send_func, *args, **kwargs): argument