Searched refs:LookupTensorSlices (Results 1 – 3 of 3) sorted by relevance
236 Status LookupTensorSlices(StringPiece key, std::vector<TensorSlice>* slices)
362 TF_ASSERT_OK(reader.LookupTensorSlices("foo", &slices)); in TEST()
911 Status BundleReader::LookupTensorSlices(StringPiece key, in LookupTensorSlices() function in tensorflow::BundleReader