| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 234 NodeMap node_map(&output); in TEST_F() local
269 NodeMap node_map(&output); in TEST_F() local
290 NodeMap node_map(&output); in TEST_F() local
303 NodeMap node_map(&output); in TEST_F() local
316 NodeMap node_map(&output); in TEST_F() local
329 NodeMap node_map(&output); in TEST_F() local
342 NodeMap node_map(&output); in TEST_F() local
355 NodeMap node_map(&output); in TEST_F() local
370 NodeMap node_map(&output); in TEST_F() local
459 NodeMap node_map(&output); in TEST_F() local
[all …]
|
| D | graph_optimizer_stage_test.cc | 92 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
|
| D | scoped_allocator_optimizer.cc | 126 Status GetInputs(NodeMap* node_map, const std::vector<NodeDef*>& ops, in GetInputs()
162 void RemoveNode(NodeDef* nd, GraphDef* graph, NodeMap* node_map) { in RemoveNode()
180 NodeDef* to_node, NodeMap* node_map) { in RemoveEdge()
258 NodeMap* node_map) { in ClearInternalControlInputs()
279 Status AnalyzeInputs(ScopedAllocatorOptimizer* sa_opti, NodeMap* node_map, in AnalyzeInputs()
314 ScopedAllocatorOptimizer* sa_opti, GraphDef* graph, NodeMap* node_map, in ConstructScopedAllocatorNode()
350 Status BuildSAConcatNode(GraphDef* graph, NodeMap* node_map, in BuildSAConcatNode()
409 Status BuildReplacementOp(GraphDef* graph, NodeMap* node_map, in BuildReplacementOp()
436 Status BuildSplitNode(GraphDef* graph, NodeMap* node_map, in BuildSplitNode()
469 Status RewireSubgraph(GraphDef* graph, NodeMap* node_map, in RewireSubgraph()
[all …]
|
| D | model_pruner.cc | 185 const NodeMap& node_map, const std::vector<string>& terminal_nodes, in IdentityNTerminalPorts()
258 GraphDef* graph, NodeMap* node_map) { in NewIdentityFromIdentityN()
283 NodeMap* node_map) { in RewriteIdentityNAndInputsOutputs()
376 NodeMap node_map(graph); in SplitIdentityNInputs() local
|
| D | arithmetic_optimizer_test.cc | 118 NodeMap node_map(&output); in TEST_F() local
158 NodeMap node_map(&output); in TEST_F() local
190 NodeMap node_map(&output); in TEST_F() local
233 NodeMap node_map(&output); in TEST_F() local
371 NodeMap node_map(&output); in TEST_F() local
415 NodeMap node_map(&output); in TEST_F() local
476 NodeMap node_map(&output); in TEST_F() local
563 NodeMap node_map(&output); in TEST_F() local
643 NodeMap node_map(&output); in TEST_F() local
696 NodeMap node_map(&output); in TEST_F() local
[all …]
|
| D | scoped_allocator_optimizer.h | 66 NodeMap* node_map() { return node_map_.get(); } in node_map() function
|
| D | memory_optimizer.cc | 72 const NodeMap& node_map, const GraphDef* graph, in FindCandidateRecomputeNodes()
109 void connected_subgraph(const NodeMap& node_map, bool collect_inputs, in connected_subgraph()
152 const GraphDef* graph, const NodeMap& node_map, in GetOpGroupsToRecompute()
215 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in GetMaxDownstreamComponents()
283 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in AddRecomputeControlDependencyNodes()
377 const std::unordered_set<NodeDef*>& target_nodes, const NodeMap& node_map, in RecomputeSubgraph()
440 NodeMap node_map(graph); in RecomputationRewritingPass() local
|
| D | loop_optimizer.cc | 521 NodeMap node_map(optimized_graph); in RemoveStackOps() local
596 const NodeDef& switch_node, const NodeMap& node_map, in CheckForDeadFanout()
724 NodeMap node_map(optimized_graph); in Optimize() local
738 const NodeMap& node_map, const absl::flat_hash_set<string>& feed_nodes, in RemoveDeadBranches()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 66 Status AddArgNodes(Graph* graph, const NodeMap& node_map, in AddArgNodes()
122 Status AddRetvalNodes(Graph* graph, const NodeMap& node_map, in AddRetvalNodes()
158 NodeMap node_map; in RewriteAndPruneGraph() local
|
| D | functionalize_while.cc | 99 std::vector<Node*>* node_map, Graph* output) { in CopySubgraph()
152 std::vector<Node*> node_map(graph.num_node_ids(), nullptr); in BuildLoopCondition() local
189 std::vector<Node*> node_map(graph.num_node_ids(), nullptr); in BuildLoopBody() local
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | colocation.cc | 78 NodeMap node_map(graph); in ReassignColocation() local
|
| D | grappler_test_test.cc | 60 NodeMap node_map(&graph); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_lib.py | 157 def node_from_map(node_map, name): argument
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_test.cc | 253 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
|
| D | transform_utils_test.cc | 54 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
|
| D | quantize_nodes_test.cc | 550 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
|
| D | freeze_requantization_ranges_test.cc | 126 std::map<string, const NodeDef*> node_map; in TestFreezeRequantizationRanges() local
|
| D | fuse_quantized_convolution.cc | 37 std::map<string, const NodeDef*> node_map; in FuseQuantizedConvolutionAndRequantize() local
|
| D | freeze_requantization_ranges.cc | 128 std::map<string, const NodeDef*> node_map; in FreezeRequantizationRanges() local
|
| D | transform_utils.cc | 72 std::map<string, const NodeDef*> node_map; in MapNodesToOutputs() local
557 std::map<string, const NodeDef*> node_map; in FindInvalidInputs() local
574 std::map<string, const NodeDef*> node_map; in IsGraphValid() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 359 Node* n, std::unordered_map<Node*, std::vector<Node*>>* node_map, in AddNodeToConstantGraph()
395 std::unordered_map<Node*, std::vector<Node*>>* node_map, in AddShapeNodeToConstantGraph()
425 std::unordered_map<Node*, std::vector<Node*>> node_map; in GetConstantGraph() local
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.cc | 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()
|
| D | utils_test.cc | 248 NodeMap node_map(&graph); in TEST_F() local
350 NodeMap node_map(&graph); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 89 std::unordered_map<string, Node*>* node_map) { in BuildNodeMap()
198 const std::unordered_map<string, Node*>& node_map, in GetEngineInfo()
831 std::unordered_map<string, Node*> node_map; in ConvertAfterShapes() local
|
| /external/v8/src/compiler/ |
| D | loop-peeling.cc | 108 NodeMarker<size_t> node_map; member
|