Home
last modified time | relevance | path

Searched refs:LookupTensorShape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsave_v2_op_test.cc132 TF_EXPECT_OK(reader.LookupTensorShape("tensor_bool", &shape)); in TEST_F()
148 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int", &shape)); in TEST_F()
164 TF_EXPECT_OK(reader.LookupTensorShape("tensor_float", &shape)); in TEST_F()
180 TF_EXPECT_OK(reader.LookupTensorShape("tensor_double", &shape)); in TEST_F()
196 TF_EXPECT_OK(reader.LookupTensorShape("tensor_qint8", &shape)); in TEST_F()
212 TF_EXPECT_OK(reader.LookupTensorShape("tensor_qint32", &shape)); in TEST_F()
229 TF_EXPECT_OK(reader.LookupTensorShape("tensor_uint8", &shape)); in TEST_F()
245 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int8", &shape)); in TEST_F()
261 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int16", &shape)); in TEST_F()
277 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int64", &shape)); in TEST_F()
[all …]
Dsave_restore_tensor.cc250 if (!reader->LookupTensorShape(tensor_name, &restored_full_shape).ok()) { in should_run_in_pool()
271 reader->LookupTensorShape(tensor_name, &restored_full_shape)); in run()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.h204 Status LookupTensorShape(StringPiece key,
Dtensor_bundle.cc1059 Status BundleReader::LookupTensorShape(StringPiece key, TensorShape* shape) { in LookupTensorShape() function in tensorflow::BundleReader