Home
last modified time | relevance | path

Searched refs:tensor_slices_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc772 gtl::STLDeleteValues(&tensor_slices_); in ~BundleReader()
942 gtl::FindPtrOrNull(tensor_slices_, full_tensor_key_string); in GetSliceValue()
953 /* full slice */ TensorSlice(full_shape.dims()), &tensor_slices_)); in GetSliceValue()
958 /* tag */ "", TensorSlice(slice), &tensor_slices_)); in GetSliceValue()
960 tss = gtl::FindPtrOrNull(tensor_slices_, full_tensor_key_string); in GetSliceValue()
Dtensor_bundle.h297 std::unordered_map<string, checkpoint::TensorSliceSet*> tensor_slices_; variable