Searched refs:tensor_slices_ (Results 1 – 2 of 2) sorted by relevance
815 for (auto& temp : tensor_slices_) { in ~BundleReader()819 tensor_slices_.clear(); in ~BundleReader()1002 gtl::FindPtrOrNull(tensor_slices_, full_tensor_key_string); in GetSliceValue()1013 /* full slice */ TensorSlice(full_shape.dims()), &tensor_slices_)); in GetSliceValue()1018 /* tag */ "", TensorSlice(slice), &tensor_slices_)); in GetSliceValue()1020 tss = gtl::FindPtrOrNull(tensor_slices_, full_tensor_key_string); in GetSliceValue()
299 std::unordered_map<string, checkpoint::TensorSliceSet*> tensor_slices_; variable