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

Method LookupSlice

tensorflow/core/util/tensor_bundle/tensor_bundle.cc:1207–1213  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1205}
1206
1207Status BundleReader::LookupSlice(StringPiece full_tensor_key,
1208 const TensorSlice& slice_spec, Tensor* val) {
1209 CHECK(val != nullptr);
1210 BundleEntryProto entry;
1211 TF_RETURN_IF_ERROR(GetBundleEntryProto(full_tensor_key, &entry));
1212 return GetSliceValue(full_tensor_key, entry, slice_spec, val);
1213}
1214
1215Status BundleReader::LookupTensorSliceProtos(
1216 StringPiece key, std::vector<TensorSliceProto>* slices) {

Callers 4

ReadTensorFromCheckpointFunction · 0.80
ComputeMethod · 0.80
runMethod · 0.80
TESTFunction · 0.80

Calls

no outgoing calls

Tested by 1

TESTFunction · 0.64