Home
last modified time | relevance | path

Searched refs:tflite_tensor_index (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dop_builder.cc161 bool GraphBuilder::HasTensor(int tflite_tensor_index) { in HasTensor() argument
162 if (tensors_.size() <= tflite_tensor_index) { in HasTensor()
165 return tensors_[tflite_tensor_index].NodeID() != -1; in HasTensor()
168 bool GraphBuilder::IsTensorUsed(int tflite_tensor_index) { in IsTensorUsed() argument
169 if (!HasTensor(tflite_tensor_index)) return false; in IsTensorUsed()
170 return used_tensor_[tflite_tensor_index]; in IsTensorUsed()
Dop_builder.h79 bool HasTensor(int tflite_tensor_index);
82 bool IsTensorUsed(int tflite_tensor_index);
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dop_builder.h270 OpBuilder::TensorID GetHexagonTensorId(int tflite_tensor_index) { in GetHexagonTensorId() argument
271 if (!HasTensor(tflite_tensor_index)) { in GetHexagonTensorId()
275 return tensors_[tflite_tensor_index]; in GetHexagonTensorId()
279 bool HasTensor(int tflite_tensor_index) { in HasTensor() argument
280 if (tensors_.size() <= tflite_tensor_index) { in HasTensor()
285 return tensors_[tflite_tensor_index].first != 0; in HasTensor()