Searched refs:supported_nodes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate.cc | 97 std::vector<int> supported_nodes = helper.GetNodesOfFirstNLargestPartitions( in DelegatePrepare() local
104 delegate->Name(), supported_nodes.size(), in DelegatePrepare()
111 BuildTfLiteIntArray(supported_nodes).get(), base_delegate); in DelegatePrepare()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | utils_test.cc | 230 std::vector<int> supported_nodes = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}; in TEST() local
231 GraphPartitionHelper helper(&mocked_context, supported_nodes); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.h | 305 const std::vector<int>& supported_nodes,
|
| D | nnapi_delegate_kernel.h | 310 TfLiteContext* context, std::vector<int>* supported_nodes,
|
| D | nnapi_delegate.cc | 3786 TfLiteContext* context, std::vector<int>* supported_nodes, in GetOperationsSupportedByTargetNnApiDevices() argument
3817 supported_nodes->clear(); in GetOperationsSupportedByTargetNnApiDevices()
3819 [&supported_nodes, &tflite_ops_support_status](int node_index) { in GetOperationsSupportedByTargetNnApiDevices()
3821 supported_nodes->push_back(node_index); in GetOperationsSupportedByTargetNnApiDevices()
4953 const std::vector<int>& supported_nodes, in GetNodesSupportedByAccelerator() argument
4959 auto supported_nodes_int_array = BuildTfLiteIntArray(supported_nodes); in GetNodesSupportedByAccelerator()
4989 if (device_supported_nodes->size() != supported_nodes.size()) { in GetNodesSupportedByAccelerator()
5100 std::vector<int> supported_nodes; in DoPrepare() local
5117 supported_nodes.push_back(node_index); in DoPrepare()
5122 if (supported_nodes.empty()) { in DoPrepare()
[all …]
|