Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc240 NodeMap node_map(&output); in TEST_F() local
275 NodeMap node_map(&output); in TEST_F() local
296 NodeMap node_map(&output); in TEST_F() local
309 NodeMap node_map(&output); in TEST_F() local
322 NodeMap node_map(&output); in TEST_F() local
335 NodeMap node_map(&output); in TEST_F() local
348 NodeMap node_map(&output); in TEST_F() local
361 NodeMap node_map(&output); in TEST_F() local
374 NodeMap node_map(&output); in TEST_F() local
390 NodeMap node_map(&output); in TEST_F() local
[all …]
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()
377 NodeMap node_map(&optimized_graph); in TEST_F() local
494 NodeMap node_map(&optimized_graph); in TEST_F() local
527 NodeMap node_map(&optimized_graph); in TEST_F() local
589 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.cc131 NodeMap node_map(&output); in TEST_F() local
280 NodeMap node_map(&output); in TEST_F() local
327 NodeMap node_map(&output); in TEST_F() local
392 NodeMap node_map(&output); in TEST_F() local
479 NodeMap node_map(&output); in TEST_F() local
562 NodeMap node_map(&output); in TEST_F() local
618 NodeMap node_map(&output); in TEST_F() local
658 NodeMap node_map(&output); in TEST_F() local
696 NodeMap node_map(&output); in TEST_F() local
751 NodeMap node_map(&output); in TEST_F() local
[all …]
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
Dscoped_allocator_optimizer.h76 NodeMap* node_map() { return node_map_.get(); } in node_map() function
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
763 const NodeMap& node_map, const absl::flat_hash_set<string>& feed_nodes, in RemoveDeadBranches()
/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.cc60 NodeMap node_map(&graph); in TEST_F() 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.cc64 std::vector<Node*>* node_map, Graph* output) { in CopySubgraph()
129 std::vector<Node*> node_map(graph.num_node_ids(), nullptr); in BuildLoopCondition() local
166 std::vector<Node*> node_map(graph.num_node_ids(), nullptr); in BuildLoopBody() local
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc229 bool HasRegularOutputs(const NodeDef& node, const NodeMap& node_map) { in HasRegularOutputs()
243 bool HasControlOutputs(const NodeDef& node, const NodeMap& node_map) { in HasControlOutputs()
258 int NumControlOutputs(const NodeDef& node, const NodeMap& node_map) { in NumControlOutputs()
274 int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map) { in NumNonControlOutputs()
294 int NumNonControlDataOutputs(const NodeDef& node, const NodeMap& node_map) { in NumNonControlDataOutputs()
323 NodeDef* GetTailOfChain(const NodeDef& source, const NodeMap& node_map, in GetTailOfChain()
Dutils_test.cc250 NodeMap node_map(&graph); in TEST_F() local
354 NodeMap node_map(&graph); in TEST_F() local
391 ImmutableNodeMap node_map(&graph); in TEST_F() local
695 NodeMap node_map(&graph); in BM_NodeMapConstruct() local
705 ImmutableNodeMap node_map(&graph); in BM_ImmutableNodeMapConstruct() 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
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_lib.py161 def node_from_map(node_map, name): argument
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils_test.cc43 std::unordered_map<string, Node*> node_map = graph.BuildNodeNameIndex(); in VerifyPlacerInspectionRequiredOps() local
Dconstant_folding.cc386 Node* n, std::unordered_map<Node*, std::vector<Node*>>* node_map, in AddNodeToConstantGraph()
422 std::unordered_map<Node*, std::vector<Node*>>* node_map, in AddShapeNodeToConstantGraph()
452 std::unordered_map<Node*, std::vector<Node*>> node_map; in GetConstantGraph() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc73 std::unordered_map<string, Node*>* node_map) { in BuildNodeMap()
133 const std::unordered_map<string, Node*>& node_map, in GetEngineInfo()
756 std::unordered_map<string, Node*> node_map; in ConvertAfterShapes() local

12