Searched refs:tensor_slices_ (Results 1 – 2 of 2) sorted by relevance
772 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()
297 std::unordered_map<string, checkpoint::TensorSliceSet*> tensor_slices_; variable