Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils_test.cc88 std::map<string, std::vector<const NodeDef*>> outputs_map; in TestMapNodesToOutputs() local
89 MapNodesToOutputs(graph_def, &outputs_map); in TestMapNodesToOutputs()
91 EXPECT_EQ(1, outputs_map.count("a")); in TestMapNodesToOutputs()
92 EXPECT_EQ(1, outputs_map["a"].size()); in TestMapNodesToOutputs()
93 EXPECT_EQ("add", outputs_map["a"][0]->name()); in TestMapNodesToOutputs()
95 EXPECT_EQ(1, outputs_map.count("b")); in TestMapNodesToOutputs()
96 EXPECT_EQ(1, outputs_map["b"].size()); in TestMapNodesToOutputs()
97 EXPECT_EQ("add", outputs_map["b"][0]->name()); in TestMapNodesToOutputs()
99 EXPECT_EQ(1, outputs_map.count("add")); in TestMapNodesToOutputs()
100 EXPECT_EQ(1, outputs_map["add"].size()); in TestMapNodesToOutputs()
[all …]
Dtransform_utils.cc380 std::map<string, std::vector<const NodeDef*>> outputs_map; in ReplaceMatchingOpTypes() local
381 MapNodesToOutputs(input_graph_def, &outputs_map); in ReplaceMatchingOpTypes()
417 if (outputs_map.count(matched_node.name())) { in ReplaceMatchingOpTypes()
419 outputs_map[matched_node.name()]) { in ReplaceMatchingOpTypes()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.cc226 std::unordered_map<int, std::tuple<void*, int>> outputs_map; in SetupGraph() local
238 CHECK(outputs_map.count(node_id) == 0); in SetupGraph()
239 outputs_map.emplace(node_id, std::make_tuple(outputs_ptr, count)); in SetupGraph()
274 CHECK(outputs_map.count(node_id) <= 1); in SetupGraph()
276 const bool has_output = outputs_map.count(node_id) == 1; in SetupGraph()
283 const auto& output_ptr_and_count = outputs_map.at(node_id); in SetupGraph()