Searched refs:recv_op (Results 1 – 3 of 3) sorted by relevance
64 RecvTPUEmbeddingActivationsOp recv_op; in runOnFunction() local65 if (failed(GetOp(func, &recv_op))) return signalPassFailure(); in runOnFunction()71 if (!recv_op && !send_op) return; in runOnFunction()73 Location loc = recv_op ? recv_op.getLoc() : send_op.getLoc(); in runOnFunction()74 StringRef config = recv_op ? recv_op.config() : send_op.config(); in runOnFunction()83 if (recv_op) in runOnFunction()84 AddOperandAndRewriteAs<_RecvTPUEmbeddingActivationsOp>(recv_op, dedup_op, in runOnFunction()
519 const NodeDef* recv_op = it->second; in Init() local521 curr_node_state.inputs.push_back(std::make_pair(recv_op, 0)); in Init()522 auto& input_node_state = node_map_.at(recv_op); in Init()
300 const string recv_op = (host_memory) ? "_HostRecv" : "_Recv"; in AddRecv() local301 NodeDefBuilder recv_builder(opts.new_name(src->name()), recv_op, in AddRecv()