| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_op_test.cc | 140 EXPECT_TRUE(reader.HasTensor("tensor_bool", &shape, &type)); in TEST_F()
159 EXPECT_TRUE(reader.HasTensor("tensor_int", &shape, &type)); in TEST_F()
178 EXPECT_TRUE(reader.HasTensor("tensor_float", &shape, &type)); in TEST_F()
197 EXPECT_TRUE(reader.HasTensor("tensor_double", &shape, &type)); in TEST_F()
216 EXPECT_TRUE(reader.HasTensor("tensor_qint8", &shape, &type)); in TEST_F()
234 EXPECT_TRUE(reader.HasTensor("tensor_qint32", &shape, &type)); in TEST_F()
252 EXPECT_TRUE(reader.HasTensor("tensor_uint8", &shape, &type)); in TEST_F()
270 EXPECT_TRUE(reader.HasTensor("tensor_int8", &shape, &type)); in TEST_F()
288 EXPECT_TRUE(reader.HasTensor("tensor_int16", &shape, &type)); in TEST_F()
306 EXPECT_TRUE(reader.HasTensor("tensor_int64", &shape, &type)); in TEST_F()
[all …]
|
| D | save_restore_tensor.cc | 181 context, reader->HasTensor(tensor_name, &saved_shape, &type), in RestoreTensor()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader_test.cc | 124 EXPECT_TRUE(reader.HasTensor("test", &shape, &type)); in SimpleFloatHelper()
127 EXPECT_FALSE(reader.HasTensor("don't exist", nullptr, nullptr)); in SimpleFloatHelper()
249 EXPECT_TRUE(reader.HasTensor("test", &shape, &type)); in SimpleIntXHelper()
252 EXPECT_FALSE(reader.HasTensor("don't exist", nullptr, nullptr)); in SimpleIntXHelper()
383 EXPECT_TRUE(reader->HasTensor("test", &shape, &type)); in TEST_SIMPLE_INT()
386 EXPECT_FALSE(reader->HasTensor("don't exist", nullptr, nullptr)); in TEST_SIMPLE_INT()
|
| D | tensor_slice_reader.h | 84 bool HasTensor(const string& name, TensorShape* shape, DataType* type) const;
|
| D | tensor_slice_reader.cc | 201 bool TensorSliceReader::HasTensor(const string& name, TensorShape* shape, in HasTensor() function in tensorflow::checkpoint::TensorSliceReader
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map_test.cc | 102 EXPECT_FALSE(buffer_map.HasTensor(0)); in TEST()
111 ASSERT_TRUE(buffer_map.HasTensor(0)); in TEST()
129 ASSERT_TRUE(buffer_map.HasTensor(0)); in TEST()
|
| D | buffer_map.cc | 135 bool BufferMap::HasTensor(int tensor_index) const { in HasTensor() function in tflite::flex::BufferMap
140 return HasTensor(tensor_index) && owned_by_tf_.count(tensor_index) > 0; in IsTensorFlowTensor()
|
| D | buffer_map.h | 39 bool HasTensor(int tensor_index) const;
|
| D | kernel.cc | 283 if (!buffer_map->HasTensor(input_index)) { in BuildEagerInputs()
468 if (!buffer_map->HasTensor(tensor_index)) { in Prepare()
541 if (!buffer_map->HasTensor(tensor_index)) { in Eval()
|
| D | delegate.cc | 82 if (!buffer_map->HasTensor(buffer_handle)) { in CopyFromBufferHandle()
|
| D | kernel_test.cc | 84 if (!buffer_map->HasTensor(buffer_handle)) { in ConfigureDelegate()
|
| /external/tensorflow/tensorflow/c/ |
| D | checkpoint_reader.cc | 65 bool CheckpointReader::HasTensor(const string& name) const { in HasTensor() function in tensorflow::checkpoint::CheckpointReader
67 return reader_->HasTensor(name, nullptr, nullptr); in HasTensor()
|
| D | checkpoint_reader.h | 44 bool HasTensor(const string& name) const;
|
| /external/tensorflow/tensorflow/python/util/ |
| D | py_checkpoint_reader.i | 138 %rename("_HasTensor") tensorflow::checkpoint::CheckpointReader::HasTensor;
|