Home
last modified time | relevance | path

Searched refs:GetOutputs (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver_test.cc34 ASSERT_THAT(runner->GetOutputs(), ElementsAre(5, 6)); in TEST()
71 ASSERT_THAT(runner->GetOutputs(), ElementsAre(5, 6)); in TEST()
106 ASSERT_THAT(runner->GetOutputs(), ElementsAre(2)); in TEST()
Dtf_driver.h47 const std::vector<int>& GetOutputs() override { return output_ids_; } in GetOutputs() function
Dtest_runner_test.cc28 const std::vector<int>& GetOutputs() override { return ids_; } in GetOutputs() function in tflite::testing::__anonc1c12f320111::ConcreteTestRunner
Dtest_runner.h43 virtual const std::vector<int>& GetOutputs() = 0;
Dtflite_driver.h61 const std::vector<int>& GetOutputs() override { in GetOutputs() function
Dtf_driver_test.cc100 ASSERT_THAT(runner->GetOutputs(), ElementsAre(0, 1)); in TEST()
Dgenerate_testspec.cc135 auto outputs = runner->GetOutputs(); in GenerateTestSpecFromRunner()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer_test.cc381 auto& nd_set = node_map.GetOutputs(nd->name()); in TEST_F()
391 auto& nd_set = node_map.GetOutputs("scoped_allocator_concat_1_1"); in TEST_F()
398 auto& nd_set = node_map.GetOutputs("scoped_allocator_1_1_Abs"); in TEST_F()
405 auto& nd_set = node_map.GetOutputs("scoped_allocator_split_1_1"); in TEST_F()
533 const auto& ctl1_outputs = node_map.GetOutputs("ctl1"); in TEST_F()
539 const auto& ctl2_outputs = node_map.GetOutputs("ctl2"); in TEST_F()
548 const auto& sa_concat_outputs = node_map.GetOutputs(sa_concat->name()); in TEST_F()
555 const auto& fused_abs_outputs = node_map.GetOutputs(fused_abs->name()); in TEST_F()
590 for (NodeDef* sa_output : node_map.GetOutputs(sa_node->name())) { in TEST_F()
Dloop_optimizer.cc88 auto consumers = node_map_->GetOutputs(node->name()); in HandleInvariantEnter()
142 auto consumers = node_map_->GetOutputs(node->name()); in HandleConst()
169 for (auto* node : node_map_->GetOutputs(loop_cond_name)) { in HandleConst()
210 auto consumers = node_map_->GetOutputs(node->name()); in HandleInvariantNode()
281 auto& consumers = node_map_->GetOutputs(node->name()); in RevertInvariantNodes()
321 for (auto* consumer : node_map_->GetOutputs(node->name())) { in RevertInvariantNodes()
340 auto consumers = node_map_->GetOutputs(node->name()); in FindInvariantNodes()
354 std::make_pair(producer, node_map_->GetOutputs(name).size())); in FindInvariantNodes()
Ddependency_optimizer.cc79 for (const auto& consumer : node_map_->GetOutputs(node.name())) { in SafeToRemoveIdentity()
258 const auto output_nodes = node_map_->GetOutputs(node_name); in OptimizeNode()
281 if (node_map_->GetOutputs(node_name).empty() && fetch_nodes_known_ && in OptimizeNode()
386 const auto& output_node_set = node_map_->GetOutputs(node_name); in OptimizeNode()
432 if (node_map_->GetOutputs(input->name()).count(consumer) == 0) { in OptimizeNode()
Dconstant_folding.cc225 for (const NodeDef* output : node_map->GetOutputs(node.name())) { in AddControlDependency()
266 const auto& tmp = node_map_->GetOutputs(node->name()); in ForwardInputs()
512 auto fanouts = node_map_->GetOutputs(shape_n_node->name()); in MaterializeShapes()
710 const auto outputs = node_map_->GetOutputs(node.name()); in MaterializeBroadcastGradientArgs()
774 for (const NodeDef* fanout : node_map_->GetOutputs(node->name())) { in MaterializeReductionIndices()
1114 const auto& outputs = node_map_->GetOutputs(node.name()); in MaybeFoldable()
1456 auto outputs = node_map_->GetOutputs(node->name()); in FoldMergeNode()
1567 auto outputs = node_map_->GetOutputs(node->name()); in FoldNode()
1596 outputs = node_map_->GetOutputs(node->name()); in FoldNode()
1649 const auto& fanout = node_map_->GetOutputs(output->node(i).name()); in FoldGraph()
[all …]
Dmemory_optimizer.cc100 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in FindCandidateRecomputeNodes()
155 for (const NodeDef* output : node_map.GetOutputs(current_node->name())) { in connected_subgraph()
197 node_map.GetOutputs(unpruned_recompute_node->name())) { in GetOpGroupsToRecompute()
242 node_map.GetOutputs(original_recompute_node->name())) { in GetMaxDownstreamComponents()
277 node_map.GetOutputs(original_recompute_node->name())) { in GetMaxDownstreamComponents()
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc230 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in HasRegularOutputs()
244 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in HasControlOutputs()
260 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in NumControlOutputs()
276 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in NumNonControlOutputs()
296 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in NumNonControlDataOutputs()
/external/tensorflow/tensorflow/go/
Dsignature.go99 for name, output := range pb.GetOutputs() {
/external/libchrome/build/android/gyp/util/
Dproguard_util.py155 def GetOutputs(self): member in ProguardCmdBuilder
189 for path in self.GetOutputs():
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dinference_profiler_stage.cc169 void* reference_ptr = reference_stage_->GetOutputs()->at(i); in Run()
170 void* test_ptr = test_stage_->GetOutputs()->at(i); in Run()
Dtflite_inference_stage.h74 const std::vector<void*>* GetOutputs() const { return &outputs_; } in GetOutputs() function
Dobject_detection_stage.cc111 const std::vector<void*>* outputs = inference_stage_->GetOutputs(); in Run()
Dimage_classification_stage.cc123 accuracy_eval_stage_->SetEvalInputs(inference_stage_->GetOutputs()->at(0), in Run()
Dtflite_inference_stage_test.cc177 uint8_t* output_tensor = static_cast<uint8_t*>(stage.GetOutputs()->at(0)); in TEST()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_runner.h45 virtual const std::vector<int>& GetOutputs() = 0;
Dtflite_driver.h44 const std::vector<int>& GetOutputs() override { in GetOutputs() function
/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dutil.h109 for (auto id : runner->GetOutputs()) { in RunKernelTest()
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dbase_task_api.h91 return engine_->GetOutputs(); in GetOutputTensors()
Dtflite_engine.h131 std::vector<const TfLiteTensor*> GetOutputs();

12