MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Send

Method Send

tensorflow/contrib/verbs/rdma.cc:1118–1133  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1116}
1117
1118void RdmaTensorResponse::Send(const Tensor& in, const TensorProto& proto,
1119 bool is_dead, const Status& status) {
1120 if (!status.ok()) {
1121 SendErrorStatus(status);
1122 return;
1123 }
1124 bool can_memcpy = DataTypeCanUseMemcpy(in.dtype());
1125 bool proto_size_changed =
1126 (!can_memcpy) && (proto.ByteSize() != rm_.tensor_bytes_);
1127 if (meta_data_changed_ || proto_size_changed) {
1128 Clone(in, proto, is_dead);
1129 SendMetaData(in, proto, is_dead);
1130 } else {
1131 SendContent(in, proto, is_dead);
1132 }
1133}
1134
1135bool RdmaTensorResponse::TensorMetaDataChanged(const Tensor& in, bool is_dead) {
1136 return (rm_.data_type_ != in.dtype()) || (rm_.tensor_shape_ != in.shape()) ||

Callers 2

ReadRowMethod · 0.45
ReadSchemaMethod · 0.45

Calls 8

DataTypeCanUseMemcpyFunction · 0.85
MessageTypeToStringFunction · 0.85
ByteSizeMethod · 0.80
EnqueueItemMethod · 0.80
SendNextItemMethod · 0.80
okMethod · 0.45
dtypeMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected