Searched refs:CreateCPUBuffer (Results 1 – 2 of 2) sorted by relevance
501 void CreateCPUBuffer(size_t size, bool lock = true);
625 tx_message_buffer_->CreateCPUBuffer(RdmaMessage::kRdmaMessageBufferSize); in RdmaChannel()626 rx_message_buffer_->CreateCPUBuffer(RdmaMessage::kRdmaMessageBufferSize); in RdmaChannel()780 void RdmaMessageBuffer::CreateCPUBuffer(size_t size, bool lock) { in CreateCPUBuffer() function in tensorflow::RdmaMessageBuffer