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

Method AsProto

tensorflow/core/framework/resource_handle.cc:41–52  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

39ResourceHandle::~ResourceHandle() {}
40
41void ResourceHandle::AsProto(ResourceHandleProto* proto) const {
42 proto->set_device(device());
43 proto->set_container(container());
44 proto->set_name(name());
45 proto->set_hash_code(hash_code());
46 proto->set_maybe_type_name(maybe_type_name());
47 for (const auto& dtype_and_shape_pair : dtypes_and_shapes_) {
48 auto dtype_and_shape = proto->add_dtypes_and_shapes();
49 dtype_and_shape->set_dtype(dtype_and_shape_pair.dtype);
50 dtype_and_shape_pair.shape.AsProto(dtype_and_shape->mutable_shape());
51 }
52}
53
54Status ResourceHandle::FromProto(const ResourceHandleProto& proto) {
55 set_device(proto.device());

Callers 1

EncodeResourceHandleListFunction · 0.45

Calls 10

containerFunction · 0.85
nameFunction · 0.85
set_containerMethod · 0.80
set_hash_codeMethod · 0.80
set_maybe_type_nameMethod · 0.80
deviceFunction · 0.50
set_deviceMethod · 0.45
set_nameMethod · 0.45
set_dtypeMethod · 0.45
mutable_shapeMethod · 0.45

Tested by

no test coverage detected