Searched refs:TensorKey (Results 1 – 1 of 1) sorted by relevance
157 using TensorKey = uint32_t; typedef in tensorflow::__anon0ef87b4d0111::GdrMemoryManager158 std::atomic<TensorKey> next_key_;162 std::map<TensorKey, const TensorBuffer*> tensor_buffers_ GUARDED_BY(buf_mu_);171 std::map<TensorKey, StatusCallback> tensor_callbacks_305 TensorKey tensor_key = ntohl(wc[i].imm_data); in Run()336 TensorKey key = wc[i].wr_id; in Run()391 TensorKey tensor_key = next_key_++; in TransportOptionsFromTensor()476 TensorKey tensor_key = remote_mr.tensor_key(); in TensorFromTransportOptions()