Searched refs:SetTensorInRecvBufResp (Results 1 – 1 of 1) sorted by relevance
540 void SetTensorInRecvBufResp(int64 max_chunk_bytes, const Tensor* tensor, in SetTensorInRecvBufResp() function601 SetTensorInRecvBufResp(recv_buf_max_chunk_, cpu_tensor, in RecvBufAsync()613 SetTensorInRecvBufResp(recv_buf_max_chunk_, hook->prod_value, in RecvBufAsync()