Searched refs:LookupDtypeAndShape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.h | 198 Status LookupDtypeAndShape(StringPiece key, DataType* dtype,
|
| D | tensor_bundle_test.cc | 67 TF_ASSERT_OK(reader->LookupDtypeAndShape(key, &dtype, &shape)); in Expect()
84 TF_ASSERT_OK(reader->LookupDtypeAndShape(key, &dtype, &shape)); in ExpectVariant()
526 TF_ASSERT_OK(reader.LookupDtypeAndShape("long_scalar", &dtype, &shape)); in TEST()
|
| D | tensor_bundle.cc | 1050 Status BundleReader::LookupDtypeAndShape(StringPiece key, DataType* dtype, in LookupDtypeAndShape() function in tensorflow::BundleReader
1061 return LookupDtypeAndShape(key, &ignored, shape); in LookupTensorShape()
|
| /external/tensorflow/tensorflow/c/ |
| D | checkpoint_reader.cc | 98 status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape); in GetTensor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 139 OP_REQUIRES_OK(context, reader.LookupDtypeAndShape( in Compute()
|
| D | save_restore_tensor.cc | 345 TF_RETURN_IF_ERROR(default_reader.LookupDtypeAndShape( in RestoreTensorsV2()
|