| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_op_test.cc | 143 EXPECT_TRUE(reader.HasTensor("tensor_bool", &shape, &type)); in TEST_F()
162 EXPECT_TRUE(reader.HasTensor("tensor_int", &shape, &type)); in TEST_F()
181 EXPECT_TRUE(reader.HasTensor("tensor_float", &shape, &type)); in TEST_F()
200 EXPECT_TRUE(reader.HasTensor("tensor_double", &shape, &type)); in TEST_F()
219 EXPECT_TRUE(reader.HasTensor("tensor_qint8", &shape, &type)); in TEST_F()
237 EXPECT_TRUE(reader.HasTensor("tensor_qint32", &shape, &type)); in TEST_F()
255 EXPECT_TRUE(reader.HasTensor("tensor_uint8", &shape, &type)); in TEST_F()
273 EXPECT_TRUE(reader.HasTensor("tensor_int8", &shape, &type)); in TEST_F()
291 EXPECT_TRUE(reader.HasTensor("tensor_int16", &shape, &type)); in TEST_F()
309 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 | 206 bool TensorSliceReader::HasTensor(const string& name, TensorShape* shape, in HasTensor() function in tensorflow::checkpoint::TensorSliceReader
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | op_builder.cc | 149 if (!HasTensor(tensor_id)) { in GetTensorID()
161 bool GraphBuilder::HasTensor(int tflite_tensor_index) { in HasTensor() function in tflite::delegates::coreml::GraphBuilder
169 if (!HasTensor(tflite_tensor_index)) return false; in IsTensorUsed()
|
| D | op_builder.h | 79 bool HasTensor(int tflite_tensor_index);
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map_test.cc | 96 EXPECT_FALSE(buffer_map.HasTensor(0)); in TEST()
105 ASSERT_TRUE(buffer_map.HasTensor(0)); in TEST()
123 ASSERT_TRUE(buffer_map.HasTensor(0)); in TEST()
|
| D | buffer_map.cc | 140 bool BufferMap::HasTensor(int tensor_index) const { in HasTensor() function in tflite::flex::BufferMap
145 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()
459 if (!buffer_map->HasTensor(tensor_index)) { in Prepare()
622 if (!buffer_map->HasTensor(tensor_index)) { in Eval()
|
| D | delegate.cc | 103 if (!buffer_map->HasTensor(buffer_handle)) { in CopyFromBufferHandle()
|
| /external/tensorflow/tensorflow/c/ |
| D | checkpoint_reader.cc | 64 bool CheckpointReader::HasTensor(const string& name) const { in HasTensor() function in tensorflow::checkpoint::CheckpointReader
66 return reader_->HasTensor(name, nullptr, nullptr); in HasTensor()
|
| D | checkpoint_reader.h | 44 bool HasTensor(const string& name) const;
|
| D | c_api_experimental.cc | 359 return reader->HasTensor(name); in TF_CheckpointReaderHasTensor()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | op_builder.h | 271 if (!HasTensor(tflite_tensor_index)) { in GetHexagonTensorId()
279 bool HasTensor(int tflite_tensor_index) { in HasTensor() function
319 if (!overwrite && HasTensor(tflite_tensor_id)) {
|
| D | op_builder.cc | 215 if (!HasTensor(tensor_id)) in AddConstNodeWithData()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 69 bool HasTensor(const TfLiteNode* node, const int index) { in HasTensor() function
75 return !HasTensor( in HasCifg()
82 return HasTensor( in HasPeephole()
87 return HasTensor( in HasNormalization()
93 return HasTensor(node, in HasProjection()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | py_checkpoint_reader_wrapper.cc | 147 .def("_HasTensor", &tensorflow::checkpoint::CheckpointReader::HasTensor) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 126 tensorflow::checkpoint::CheckpointReader::HasTensor
|