Searched refs:proto_size (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.cc | 1345 uint32_t* proto_size = (uint32_t*)&message[kErrorStatusStartIndex]; in CreateMessage() local
1346 *proto_size = gsProtoSize; in CreateMessage()
1462 bool is_dead, size_t proto_size) { in SetTensorMetaData() argument
1467 meta_data.proto_size_ = proto_size; in SetTensorMetaData()
1554 uint32_t proto_size = meta_data_->proto_size_; in AllocateTensors() local
1555 rdma_addr_ = malloc(proto_size); in AllocateTensors()
1556 mr_ = ibv_reg_mr(RdmaMemoryMgr::Singleton().pd_, rdma_addr_, proto_size, in AllocateTensors()
1608 bool is_dead, size_t proto_size) { in RecvTensorMetaData() argument
1610 key_, dtype, shape, is_dead, proto_size); in RecvTensorMetaData()
|
| D | rdma.h | 212 bool is_dead, size_t proto_size);
260 size_t proto_size);
|
| D | patch_notes_verbs_with_0_copies.md | 38 * **class RdmaTensorMetaData** - Meta-data for a tensor (type, shape, is_dead, proto_size).
|
| D | README.md | 107 * **class RdmaTensorMetaData** - Meta-data for a tensor (type, shape, is_dead, proto_size).
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 653 const size_t proto_size = in.ByteSizeLong(); in MessageToBuffer() local
654 void* buf = tensorflow::port::Malloc(proto_size); in MessageToBuffer()
658 in.GetTypeName(), "' and size ", proto_size); in MessageToBuffer()
664 if (proto_size > std::numeric_limits<int>::max()) { in MessageToBuffer()
667 proto_size, in MessageToBuffer()
671 if (!in.SerializeToArray(buf, proto_size)) { in MessageToBuffer()
674 proto_size, " bytes) is too large?"); in MessageToBuffer()
677 out->length = proto_size; in MessageToBuffer()
|