| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_allocator_test.cc | 112 NodeAndRegistration* node_and_registration, size_t count) { in VerifyRegistrationAndNodeAllocation() argument
114 TF_LITE_MICRO_EXPECT_NE(nullptr, node_and_registration[i].registration); in VerifyRegistrationAndNodeAllocation()
115 TF_LITE_MICRO_EXPECT_NE(nullptr, node_and_registration[i].node.inputs); in VerifyRegistrationAndNodeAllocation()
116 TF_LITE_MICRO_EXPECT_NE(nullptr, node_and_registration[i].node.outputs); in VerifyRegistrationAndNodeAllocation()
241 tflite::NodeAndRegistration* node_and_registration; in TF_LITE_MICRO_TEST() local
250 &node_and_registration, &eval_tensors)); in TF_LITE_MICRO_TEST()
254 &node_and_registration, &eval_tensors)); in TF_LITE_MICRO_TEST()
262 tflite::NodeAndRegistration* node_and_registration; in TF_LITE_MICRO_TEST() local
278 &node_and_registration, &eval_tensors)); in TF_LITE_MICRO_TEST()
282 &node_and_registration, &eval_tensors)); in TF_LITE_MICRO_TEST()
[all …]
|
| D | recording_micro_allocator_test.cc | 42 tflite::NodeAndRegistration* node_and_registration; in TF_LITE_MICRO_TEST() local
56 model, all_ops_resolver, &node_and_registration, &eval_tensors); in TF_LITE_MICRO_TEST()
86 tflite::NodeAndRegistration* node_and_registration; in TF_LITE_MICRO_TEST() local
98 model, all_ops_resolver, &node_and_registration, &eval_tensors); in TF_LITE_MICRO_TEST()
122 tflite::NodeAndRegistration* node_and_registration; in TF_LITE_MICRO_TEST() local
138 model, all_ops_resolver, &node_and_registration, &eval_tensors); in TF_LITE_MICRO_TEST()
149 model, all_ops_resolver, &node_and_registration, &eval_tensors); in TF_LITE_MICRO_TEST()
|
| D | micro_interpreter.h | 163 const NodeAndRegistration node_and_registration(int node_index) const { in node_and_registration() function
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib.cc | 95 const auto* node_and_registration = in ExportOperators() local
96 subgraph_->node_and_registration(op_index); in ExportOperators()
97 const TfLiteRegistration* registration = &node_and_registration->second; in ExportOperators()
108 const auto* node_and_registration = in ExportOperators() local
109 subgraph_->node_and_registration(op_index); in ExportOperators()
110 const TfLiteNode& node = node_and_registration->first; in ExportOperators()
111 const TfLiteRegistration& registration = node_and_registration->second; in ExportOperators()
168 const auto* node_and_registration = in ExportTensors() local
169 subgraph_->node_and_registration(op_index); in ExportTensors()
171 TfLiteIntArrayView(node_and_registration->first.temporaries)) in ExportTensors()
[all …]
|
| /external/tensorflow/tensorflow/lite/ |
| D | model_xnnpack_test.cc | 51 interpreter->node_and_registration(first_node_id)->second; in TEST()
79 interpreter->node_and_registration(first_node_id)->second; in TEST()
|
| D | optional_debug_tools.cc | 128 interpreter->node_and_registration(static_cast<int>(node_index)); in PrintInterpreterState()
|
| D | interpreter.h | 261 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration( in node_and_registration() function
263 return primary_subgraph().node_and_registration(node_index); in node_and_registration()
|
| D | model_test.cc | 235 interpreter->node_and_registration(0); in TEST()
254 interpreter->node_and_registration(1); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | test_util.mm | 46 const auto* node_and_reg = interpreter->node_and_registration(node_index);
60 const auto* node_and_reg = interpreter->node_and_registration(node_idx);
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | hexagon_delegate_op_model.h | 68 const auto* node_and_reg = interpreter_->node_and_registration(node); in ApplyDelegateAndInvoke()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.cc | 308 std::pair<TfLiteNode, TfLiteRegistration> node_and_registration) { in CountPartitionsDelegatedTo() argument
309 return node_and_registration.first.delegate == delegate; in CountPartitionsDelegatedTo()
333 std::tie(node, reg) = *(interpreter->node_and_registration(node_idx)); in CountPartitionsExecutedByCpuKernel()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 158 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration = in node() local
159 interpreter_.primary_subgraph().node_and_registration(index); in node()
160 return const_cast<TfLiteNode*>(&node_and_registration->first); in node()
163 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration = in registration() local
164 interpreter_.primary_subgraph().node_and_registration(index); in registration()
165 return const_cast<TfLiteRegistration*>(&node_and_registration->second); in registration()
|
| /external/tensorflow/tensorflow/lite/examples/label_image/ |
| D | label_image.cc | 381 const auto node_and_registration = in RunInference() local
382 subgraph->node_and_registration(op_index); in RunInference()
383 const TfLiteRegistration registration = node_and_registration->second; in RunInference()
|
| /external/tensorflow/tensorflow/lite/delegates/utils/ |
| D | simple_delegate_test.cc | 71 const auto* node_and_reg = interpreter_->node_and_registration(node); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test.cc | 57 const auto* node_and_reg = interpreter_->node_and_registration(node); in TEST_F()
266 const auto* node_and_reg = interpreter_->node_and_registration(3); in TEST_F()
361 const auto* node_and_reg = interpreter_->node_and_registration(3); in TEST_F()
785 interpreter_->subgraph(subgraph_index)->node_and_registration(node); in TEST_F()
802 interpreter_->subgraph(1)->node_and_registration(node); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 500 const TfLiteNode* node = &(interpreter_->node_and_registration(i)->first); in NodeInputs()
510 const TfLiteNode* node = &(interpreter_->node_and_registration(i)->first); in NodeOutputs()
522 &(interpreter_->node_and_registration(i)->second); in NodeName()
|
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer.cc | 75 auto node_reg = subgraph->node_and_registration(node_index); in GetOperatorDetails()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 292 const auto* node_and_reg = interpreter->node_and_registration(entry.first); in GetNodeOpInfoMapAndContext()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 203 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration( in node_and_registration() function
|
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| D | benchmark_tflite_model.cc | 649 interpreter_->node_and_registration(node_id)->first; in Init()
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | nativeinterpreterwrapper_jni.cc | 261 subgraph->node_and_registration(node_i)->second; in Java_org_tensorflow_lite_NativeInterpreterWrapper_hasUnresolvedFlexOp()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_device_selection_test.cc | 833 return interpreter_->node_and_registration(node_index) in CountNnApiPartitions()
|