Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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 …]
H A 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
H A 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 …]
H A 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
H A 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
H A 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
H A 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
H A 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()
H A Dscoped_allocator_optimizer.h73 NodeMap* node_map() { return node_map_.get(); } in node_map() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A 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()
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A 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
H A 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
H A 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
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A 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
H A 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
H A Dfunctionalize_cond.cc572 auto& node_map = node_maps_[branch_index]; in ExtractBodies() local
720 auto& node_map = node_maps_[branch_index]; in ExtractBodies() local
732 auto& node_map = node_maps_[branch_index]; in ExtractBodies() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dvirtual_scheduler.cc171 const std::unordered_map<const NodeDef*, NodeState>* node_map) { in Init()
225 const std::unordered_map<const NodeDef*, NodeState>* node_map, in FirstReadyCmp()
277 const std::unordered_map<const NodeDef*, NodeState>* node_map) { in Init()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dpasses.cpp1159 const auto& node_map = match.nodes_map; in eliminatePermuteOpsSumPattern() local
1203 const auto& node_map = match.nodes_map; in eliminatePermuteOpsSoftmaxPattern() local
1423 const auto& node_map = match.nodes_map; in FuseClampNaNToNum() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dnode_def_util_test.cc815 std::map<absl::string_view, absl::string_view> node_map; in TEST() local
830 std::map<absl::string_view, absl::string_view> node_map; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph.cc75 std::unordered_map<string, Node*>* node_map) { in BuildNodeMap()
799 std::unordered_map<string, Node*> node_map; in ConvertGraph() local
/aosp_15_r20/external/rust/android-crates-io/crates/petgraph/src/algo/
Dmin_spanning_tree.rs65 node_map: HashMap<usize, usize>, field

12