Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer_test.cc131 auto& nd_set = node_map.GetOutputs(nd->name()); in TEST_F()
141 auto& nd_set = node_map.GetOutputs("scoped_allocator_concat_1_1"); in TEST_F()
148 auto& nd_set = node_map.GetOutputs("scoped_allocator_1_1_Abs"); in TEST_F()
155 auto& nd_set = node_map.GetOutputs("scoped_allocator_split_1_1"); in TEST_F()
Dloop_optimizer.cc89 auto consumers = node_map_->GetOutputs(node->name()); in HandleInvariantEnter()
143 auto consumers = node_map_->GetOutputs(node->name()); in HandleConst()
171 for (auto* node : node_map_->GetOutputs(loop_cond_name)) { in HandleConst()
212 auto consumers = node_map_->GetOutputs(node->name()); in HandleInvariantNode()
283 auto& consumers = node_map_->GetOutputs(node->name()); in RevertInvariantNodes()
323 for (auto* consumer : node_map_->GetOutputs(node->name())) { in RevertInvariantNodes()
342 auto consumers = node_map_->GetOutputs(node->name()); in FindInvariantNodes()
356 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()
236 const std::set<NodeDef*> output_nodes = node_map_->GetOutputs(node_name); in OptimizeNode()
259 if (node_map_->GetOutputs(node_name).empty() && fetch_nodes_known_ && in OptimizeNode()
353 const auto& output_node_set = node_map_->GetOutputs(node_name); in OptimizeNode()
408 if (node_map_->GetOutputs(input->name()).count(consumer) == 0) { in OptimizeNode()
Dconstant_folding.cc253 auto outputs = node_map->GetOutputs(node->name()); in AddControlDependency()
461 auto outputs = node_map_->GetOutputs(shape_n_node->name()); in MaterializeShapes()
655 const std::set<NodeDef*> outputs = node_map_->GetOutputs(node.name()); in MaterializeBroadcastGradientArgs()
719 for (const NodeDef* fanout : node_map_->GetOutputs(node->name())) { in MaterializeReductionIndices()
927 auto outputs = node_map_->GetOutputs(node.name()); in IsFoldable()
1293 auto outputs = node_map_->GetOutputs(node->name()); in FoldMergeNode()
1395 auto outputs = node_map_->GetOutputs(node->name()); in FoldNode()
1424 outputs = node_map_->GetOutputs(node->name()); in FoldNode()
1451 const std::set<NodeDef*>& outputs = node_map_->GetOutputs(node->name()); in FoldGraph()
1479 auto fanout = node_map_->GetOutputs(output->node(i).name()); in FoldGraph()
[all …]
Dmemory_optimizer.cc81 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in FindCandidateRecomputeNodes()
136 for (const NodeDef* output : node_map.GetOutputs(current_node->name())) { in connected_subgraph()
177 for (NodeDef* output : node_map.GetOutputs(recompute_node->name())) { in GetOpGroupsToRecompute()
222 node_map.GetOutputs(original_recompute_node->name())) { in GetMaxDownstreamComponents()
257 node_map.GetOutputs(original_recompute_node->name())) { in GetMaxDownstreamComponents()
Darithmetic_optimizer.cc250 for (const NodeDef* output : ctx().node_map->GetOutputs(node.name())) { in DrivesControlDependency()
1556 const auto& outputs = ctx().node_map->GetOutputs(node.name()); in InitializeChains()
1584 if (ctx().node_map->GetOutputs(op->name()).size() > 1) { in OpsAreSafeToHoist()
1607 for (NodeDef* new_tail : ctx().node_map->GetOutputs(tail->name())) { in AdvanceTails()
1705 const std::set<NodeDef*> consumers = ctx().node_map->GetOutputs(node_name); in UpdateConsumers()
2803 ctx().node_map->GetOutputs(inner_function->name()).size() == 1 && in TrySimplify()
2836 const std::set<NodeDef*> consumers = ctx().node_map->GetOutputs(node_name); in UpdateConsumers()
3479 const std::set<NodeDef*>& tmp = node_map_->GetOutputs(node->name()); in DedupComputations()
3621 const std::set<NodeDef*> outputs = node_map_->GetOutputs(node->name()); in SimplifyArithmeticOps()
Dmodel_pruner.cc314 for (NodeDef* output : node_map->GetOutputs(node->name())) { in RewriteIdentityNAndInputsOutputs()
Dlayout_optimizer.cc555 auto outputs = node_map_->GetOutputs(node_->name()); in HasOutputs()
664 auto outputs = node_map_->GetOutputs(node_->name()); in AddTransformToOutputs()
2139 auto outputs = node_map_->GetOutputs(trans_second); in Collapse()
Dscoped_allocator_optimizer.cc478 std::set<NodeDef*> output_nodes = node_map->GetOutputs(old_op->name()); in RewireSubgraph()
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver_test.cc34 ASSERT_THAT(runner->GetOutputs(), ElementsAre(5, 6)); in TEST()
70 ASSERT_THAT(runner->GetOutputs(), ElementsAre(5, 6)); in TEST()
Dtflite_driver.h42 const std::vector<int>& GetOutputs() override { in GetOutputs() function
Dtest_runner.h42 virtual const std::vector<int>& GetOutputs() = 0;
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::__anon73a85d540111::ConcreteTestRunner
Dtf_driver_test.cc100 ASSERT_THAT(runner->GetOutputs(), ElementsAre(0, 1)); in TEST()
Dparse_testdata.cc259 expected_outputs_ = test_runner->GetOutputs(); in Invoke()
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc110 const std::set<NodeDef*>& NodeMap::GetOutputs(const string& node_name) const { in GetOutputs() function in tensorflow::grappler::NodeMap
276 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in NumNonControlOutputs()
296 for (const NodeDef* output : node_map.GetOutputs(node.name())) { in NumNonControlDataOutputs()
Dutils.h50 const std::set<NodeDef*>& GetOutputs(const string& node_name) const;
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtflite_driver.h44 const std::vector<int>& GetOutputs() override { in GetOutputs() function
Dtest_runner.h45 virtual const std::vector<int>& GetOutputs() = 0;
Dparse_testdata.cc262 expected_outputs_ = test_runner->GetOutputs(); in Invoke()
/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dutil.h109 for (auto id : runner->GetOutputs()) { in RunKernelTest()