Home
last modified time | relevance | path

Searched defs:node_map (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/utils/
Dtransitive_fanin_test.cc71 NodeMap node_map(&output_graph); in TEST_F() local
90 NodeMap node_map(&output_graph); in TEST_F() local
111 NodeMap node_map(&output_graph); in TEST_F() local
Dcolocation.cc89 NodeMap node_map(graph); in ReassignColocation() local
Dgrappler_test_test.cc61 NodeMap node_map(&graph); in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage_test.cc92 NodeMap node_map(&graph); in TEST_F() local
121 NodeMap node_map(&graph); in TEST_F() local
154 NodeMap node_map(&item.graph); in TEST_F() local
196 NodeMap node_map(&item.graph); in TEST_F() local
Dscoped_allocator_optimizer.cc151 void RemoveNode(NodeDef* nd, GraphDef* graph, NodeMap* node_map) { in RemoveNode()
169 NodeDef* to_node, NodeMap* node_map) { in RemoveEdge()
217 NodeMap* node_map, const DataType& dtype, in MaybeRewriteInput()
260 NodeMap* node_map, const std::vector<NodeDef*>& ops, in GetInputs()
314 Status GetDataInputs(GraphDef* graph, NodeMap* node_map, NodeDef* op, in GetDataInputs()
425 NodeMap* node_map) { in ClearInternalControlInputs()
448 NodeMap* node_map, const std::vector<NodeDef*>& ops, in AnalyzeInputs()
484 GraphDef* graph, NodeMap* node_map, in TransitiveFanoutWithinFrame()
511 ScopedAllocatorOptimizer* sa_opti, GraphDef* graph, NodeMap* node_map, in ConstructScopedAllocatorNode()
597 Status BuildSAConcatNode(GraphDef* graph, NodeMap* node_map, in BuildSAConcatNode()
[all …]
Dscoped_allocator_optimizer_test.cc321 void GetNode(NodeMap* node_map, const string& node_name, NodeDef** node_def) { in GetNode()
328 NodeDef* ValidateSAControlInput(GraphDef* graph, NodeMap* node_map, in ValidateSAControlInput()
347 int NumControlInputs(NodeMap* node_map, const string& node_name) { in NumControlInputs()
378 NodeMap node_map(&optimized_graph); in TEST_F() local
495 NodeMap node_map(&optimized_graph); in TEST_F() local
528 NodeMap node_map(&optimized_graph); in TEST_F() local
590 NodeMap node_map(&optimized_graph); in TEST_F() local
Dcommon_subgraph_elimination_test.cc83 NodeMap node_map(&output); in TEST_F() local
123 NodeMap node_map(&output); in TEST_F() local
154 NodeMap node_map(&output); in TEST_F() local
Dmodel_pruner.cc203 const NodeMap& node_map, const std::vector<string>& terminal_nodes, in IdentityNTerminalPorts()
276 GraphDef* graph, NodeMap* node_map) { in NewIdentityFromIdentityN()
301 NodeMap* node_map) { in RewriteIdentityNAndInputsOutputs()
394 NodeMap node_map(graph); in SplitIdentityNInputs() local
Darithmetic_optimizer_test.cc139 NodeMap node_map(&g); in TEST_F() local
182 NodeMap node_map(&g); in TEST_F() local
313 NodeMap node_map(&g); in TEST_F() local
370 NodeMap node_map(&output); in TEST_F() local
419 NodeMap node_map(&g); in TEST_F() local
489 NodeMap node_map(&g); in TEST_F() local
708 NodeMap node_map(&output); in TEST_F() local
755 NodeMap node_map(&output); in TEST_F() local
820 NodeMap node_map(&output); in TEST_F() local
907 NodeMap node_map(&output); in TEST_F() local
[all …]
Dscoped_allocator_optimizer.h73 NodeMap* node_map() { return node_map_.get(); } in node_map() function
Dmemory_optimizer.cc91 const NodeMap& node_map, const GraphDef* graph, in FindCandidateRecomputeNodes()
128 void connected_subgraph(const NodeMap& node_map, bool collect_inputs, in connected_subgraph()
171 const GraphDef* graph, const NodeMap& node_map, in GetOpGroupsToRecompute()
235 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in GetMaxDownstreamComponents()
303 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in AddRecomputeControlDependencyNodes()
397 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in RecomputeSubgraph()
454 NodeMap node_map(graph); in RecomputationRewritingPass() local
Dloop_optimizer.cc520 NodeMap node_map(optimized_graph); in RemoveStackOps() local
598 const NodeDef& switch_node, const NodeMap& node_map, in CheckForDeadFanout()
749 NodeMap node_map(optimized_graph); in Optimize() local
762 const std::unordered_set<string>& nodes_to_preserve, NodeMap& node_map, in RemoveDeadBranches()
Dcommon_subgraph_elimination.cc214 NodeMap node_map(optimized_graph); in DedupComputations() local
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc214 bool HasRegularOutputs(const NodeDef& node, const NodeMap& node_map) { in HasRegularOutputs()
228 bool HasControlOutputs(const NodeDef& node, const NodeMap& node_map) { in HasControlOutputs()
243 int NumControlOutputs(const NodeDef& node, const NodeMap& node_map) { in NumControlOutputs()
259 int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map) { in NumNonControlOutputs()
279 int NumNonControlDataOutputs(const NodeDef& node, const NodeMap& node_map) { in NumNonControlDataOutputs()
308 NodeDef* GetTailOfChain(const NodeDef& source, const NodeMap& node_map, in GetTailOfChain()
Dutils_test.cc253 NodeMap node_map(&graph); in TEST_F() local
357 NodeMap node_map(&graph); in TEST_F() local
394 ImmutableNodeMap node_map(&graph); in TEST_F() local
699 NodeMap node_map(&graph); in BM_NodeMapConstruct() local
709 ImmutableNodeMap node_map(&graph); in BM_ImmutableNodeMapConstruct() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler_util.cc54 Status AddArgNodes(Graph* graph, const NodeMap& node_map, in AddArgNodes()
114 Status AddRetvalNodes(Graph* graph, const NodeMap& node_map, in AddRetvalNodes()
151 NodeMap node_map; in RewriteAndPruneGraph() local
Dfunctionalize_while.cc62 std::vector<Node*>* node_map, Graph* output) { in CopySubgraph()
127 std::vector<Node*> node_map(graph.num_node_ids(), nullptr); in BuildLoopCondition() local
164 std::vector<Node*> node_map(graph.num_node_ids(), nullptr); in BuildLoopBody() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils_test.cc54 std::map<string, const NodeDef*> node_map; in TestMapNamesToNodes() local
250 std::map<string, const NodeDef*> node_map; in TestFilterGraphDef() local
272 std::map<string, const NodeDef*> node_map; in TestRemoveAttributes() local
546 std::map<string, const NodeDef*> node_map; in TestRenameNodeInputs() local
584 std::map<string, const NodeDef*> node_map; in TestRenameNodeInputsWithRedirects() local
657 std::map<string, const NodeDef*> node_map; in TestRenameNodeInputsWithWildcard() local
694 std::map<string, const NodeDef*> node_map; in TestRenameNodeInputsWithIgnores() local
816 std::map<string, const NodeDef*> node_map; in TestGetInOutTypes() local
Dfold_constants_test.cc253 std::map<string, const NodeDef*> node_map; in TestReplaceSendRecvs() local
287 std::map<string, const NodeDef*> node_map; in TestReplaceSendRecvsPrefixNames() local
323 std::map<string, const NodeDef*> node_map; in TestRemoveUnusedNodes() local
Dquantize_nodes_test.cc550 std::map<string, const NodeDef*> node_map; in TestRemoveRedundantQuantization() local
649 std::map<string, const NodeDef*> node_map; in TestRemoveRedundantQuantizationWithBiasAdd() local
811 std::map<string, const NodeDef*> node_map; in TestQuantizePlaceholders() local
1120 std::map<string, const NodeDef*> node_map; in TestHoistFakeQuants() local
1258 std::map<string, const NodeDef*> node_map; in TestMergeDuplicateConsts() local
1307 std::map<string, const NodeDef*> node_map; in TestMergeDuplicatesNested() local
1364 std::map<string, const NodeDef*> node_map; in TestMergeDuplicatesInOut() local
1424 std::map<string, const NodeDef*> node_map; in TestExcludeNonFloat() local
Dfreeze_requantization_ranges_test.cc126 std::map<string, const NodeDef*> node_map; in TestFreezeRequantizationRanges() local
Dtransform_utils.cc74 std::map<string, const NodeDef*> node_map; in MapNodesToOutputs() local
559 std::map<string, const NodeDef*> node_map; in FindInvalidInputs() local
576 std::map<string, const NodeDef*> node_map; in IsGraphValid() local
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_lib.py157 def node_from_map(node_map, name): argument
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils_test.cc45 std::unordered_map<string, Node*> node_map = graph.BuildNodeNameIndex(); in VerifyPlacerInspectionRequiredOps() local
Dconstant_folding.cc383 Node* n, std::unordered_map<Node*, std::vector<Node*>>* node_map, in AddNodeToConstantGraph()
419 std::unordered_map<Node*, std::vector<Node*>>* node_map, in AddShapeNodeToConstantGraph()
449 std::unordered_map<Node*, std::vector<Node*>> node_map; in GetConstantGraph() local

12