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.cc134 TF_EXPECT_OK(reader.LookupTensorShape("tensor_bool", &shape)); in TEST_F()
150 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int", &shape)); in TEST_F()
166 TF_EXPECT_OK(reader.LookupTensorShape("tensor_float", &shape)); in TEST_F()
182 TF_EXPECT_OK(reader.LookupTensorShape("tensor_double", &shape)); in TEST_F()
198 TF_EXPECT_OK(reader.LookupTensorShape("tensor_qint8", &shape)); in TEST_F()
214 TF_EXPECT_OK(reader.LookupTensorShape("tensor_qint32", &shape)); in TEST_F()
231 TF_EXPECT_OK(reader.LookupTensorShape("tensor_uint8", &shape)); in TEST_F()
247 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int8", &shape)); in TEST_F()
263 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int16", &shape)); in TEST_F()
279 TF_EXPECT_OK(reader.LookupTensorShape("tensor_int64", &shape)); in TEST_F()
[all …]
Dsave_restore_tensor.cc251 if (!reader->LookupTensorShape(tensor_name, &restored_full_shape).ok()) { in should_run_in_pool()
272 reader->LookupTensorShape(tensor_name, &restored_full_shape)); in run()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.h205 Status LookupTensorShape(StringPiece key,
Dtensor_bundle.cc1120 Status BundleReader::LookupTensorShape(StringPiece key, TensorShape* shape) { in LookupTensorShape() function in tensorflow::BundleReader