| /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 | 194 context, reader->HasTensor(tensor_name, &saved_shape, &type), in RestoreTensor()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_reader_test.cc | 130 EXPECT_TRUE(reader.HasTensor("test", &shape, &type)); in SimpleFloatHelper()
133 EXPECT_FALSE(reader.HasTensor("don't exist", nullptr, nullptr)); in SimpleFloatHelper()
255 EXPECT_TRUE(reader.HasTensor("test", &shape, &type)); in SimpleIntXHelper()
258 EXPECT_FALSE(reader.HasTensor("don't exist", nullptr, nullptr)); in SimpleIntXHelper()
381 EXPECT_TRUE(reader.HasTensor("test", nullptr, nullptr)); in TEST()
409 EXPECT_TRUE(reader.HasTensor("test", nullptr, nullptr)); in TEST()
484 EXPECT_TRUE(reader.HasTensor("test", nullptr, nullptr)); in TEST()
560 EXPECT_TRUE(reader->HasTensor("test", &shape, &type)); in CachedTensorSliceReaderTesterHelper()
563 EXPECT_FALSE(reader->HasTensor("don't exist", nullptr, nullptr)); in CachedTensorSliceReaderTesterHelper()
|
| D | tensor_slice_reader.h | 84 bool HasTensor(const string& name, TensorShape* shape, DataType* type) const;
|
| D | tensor_slice_reader.cc | 211 bool TensorSliceReader::HasTensor(const string& name, TensorShape* shape, in HasTensor() function in tensorflow::checkpoint::TensorSliceReader
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map.cc | 32 bool BufferMap::HasTensor(int tensor_index) const { in HasTensor() function in tflite::flex::BufferMap
37 return HasTensor(tensor_index) && owned_by_tf_.count(tensor_index) > 0; in IsTensorFlowTensor()
|
| D | buffer_map_test.cc | 104 EXPECT_FALSE(buffer_map.HasTensor(0)); in TEST()
113 ASSERT_TRUE(buffer_map.HasTensor(0)); in TEST()
131 ASSERT_TRUE(buffer_map.HasTensor(0)); in TEST()
|
| D | buffer_map.h | 39 bool HasTensor(int tensor_index) const;
|
| D | kernel.cc | 272 if (!buffer_map->HasTensor(input_index)) { in BuildOpKernelInputs()
560 if (!tensor->data_is_stale || !buffer_map->HasTensor(tensor_index)) { in Prepare()
720 if (!tensor->data_is_stale || !buffer_map->HasTensor(tensor_index)) { in Eval()
768 if (!buffer_map->HasTensor(tensor_index)) { in Eval()
|
| D | kernel_test.cc | 195 EXPECT_TRUE(buffer_map->HasTensor(6)); in TEST_F()
196 EXPECT_FALSE(buffer_map->HasTensor(7)); in TEST_F()
|
| D | delegate.cc | 110 if (!buffer_map->HasTensor(buffer_handle)) { in CopyFromBufferHandle()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | op_builder.cc | 151 if (!HasTensor(tensor_id)) { in GetTensorID()
163 bool GraphBuilder::HasTensor(int tflite_tensor_index) { in HasTensor() function in tflite::delegates::coreml::GraphBuilder
171 if (!HasTensor(tflite_tensor_index)) return false; in IsTensorUsed()
|
| D | op_builder.h | 80 bool HasTensor(int tflite_tensor_index);
|
| /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 | 368 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 | 101 bool HasTensor(const TfLiteNode* node, const int index) { in HasTensor() function
107 return !HasTensor( in HasCifg()
114 return HasTensor( in HasPeephole()
119 return HasTensor( in HasNormalization()
125 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 | 135 tensorflow::checkpoint::CheckpointReader::HasTensor
|