| /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 | 146 void RemoveNode(NodeDef* nd, GraphDef* graph, NodeMap* node_map) { in RemoveNode()
164 NodeDef* to_node, NodeMap* node_map) { in RemoveEdge()
212 NodeMap* node_map, const DataType& dtype, in MaybeRewriteInput()
254 NodeMap* node_map, const std::vector<NodeDef*>& ops, in GetInputs()
308 Status GetDataInputs(GraphDef* graph, NodeMap* node_map, NodeDef* op, in GetDataInputs()
405 NodeMap* node_map) { in ClearInternalControlInputs()
428 NodeMap* node_map, const std::vector<NodeDef*>& ops, in AnalyzeInputs()
463 GraphDef* graph, NodeMap* node_map, in TransitiveFanoutWithinFrame()
490 ScopedAllocatorOptimizer* sa_opti, GraphDef* graph, NodeMap* node_map, in ConstructScopedAllocatorNode()
570 Status BuildSAConcatNode(GraphDef* graph, NodeMap* node_map, in BuildSAConcatNode()
[all …]
|
| D | scoped_allocator_optimizer_test.cc | 291 void GetNode(NodeMap* node_map, const string& node_name, NodeDef** node_def) { in GetNode()
298 NodeDef* ValidateSAControlInput(GraphDef* graph, NodeMap* node_map, in ValidateSAControlInput()
317 int NumControlInputs(NodeMap* node_map, const string& node_name) { in NumControlInputs()
347 NodeMap node_map(&optimized_graph); in TEST_F() local
465 NodeMap node_map(&optimized_graph); in TEST_F() local
498 NodeMap node_map(&optimized_graph); in TEST_F() local
|
| D | model_pruner.cc | 200 const NodeMap& node_map, const std::vector<string>& terminal_nodes, in IdentityNTerminalPorts()
273 GraphDef* graph, NodeMap* node_map) { in NewIdentityFromIdentityN()
298 NodeMap* node_map) { in RewriteIdentityNAndInputsOutputs()
391 NodeMap node_map(graph); in SplitIdentityNInputs() local
|
| D | arithmetic_optimizer_test.cc | 121 NodeMap node_map(&output); in TEST_F() local
161 NodeMap node_map(&output); in TEST_F() local
192 NodeMap node_map(&output); in TEST_F() local
237 NodeMap node_map(&output); in TEST_F() local
386 NodeMap node_map(&output); in TEST_F() local
433 NodeMap node_map(&output); in TEST_F() local
497 NodeMap node_map(&output); in TEST_F() local
584 NodeMap node_map(&output); in TEST_F() local
667 NodeMap node_map(&output); in TEST_F() local
723 NodeMap node_map(&output); in TEST_F() local
[all …]
|
| D | scoped_allocator_optimizer.h | 76 NodeMap* node_map() { return node_map_.get(); } in node_map() function
|
| D | memory_optimizer.cc | 91 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
|
| D | loop_optimizer.cc | 520 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/ |
| D | transitive_fanin_test.cc | 71 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
|
| 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/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 53 Status AddArgNodes(Graph* graph, const NodeMap& node_map, in AddArgNodes()
111 Status AddRetvalNodes(Graph* graph, const NodeMap& node_map, in AddRetvalNodes()
148 NodeMap node_map; in RewriteAndPruneGraph() local
|
| D | functionalize_while.cc | 64 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/ |
| D | utils.cc | 301 bool HasRegularOutputs(const NodeDef& node, const NodeMap& node_map) { in HasRegularOutputs()
315 bool HasControlOutputs(const NodeDef& node, const NodeMap& node_map) { in HasControlOutputs()
330 int NumControlOutputs(const NodeDef& node, const NodeMap& node_map) { in NumControlOutputs()
346 int NumNonControlOutputs(const NodeDef& node, const NodeMap& node_map) { in NumNonControlOutputs()
366 int NumNonControlDataOutputs(const NodeDef& node, const NodeMap& node_map) { in NumNonControlDataOutputs()
395 NodeDef* GetTailOfChain(const NodeDef& source, const NodeMap& node_map, in GetTailOfChain()
|
| D | utils_test.cc | 250 NodeMap node_map(&graph); in TEST_F() local
352 NodeMap node_map(&graph); in TEST_F() local
622 NodeMap node_map(&graph); in BM_NodeMapConstruct() local
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_lib.py | 161 def node_from_map(node_map, name): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_inspection_required_ops_utils_test.cc | 44 std::unordered_map<string, Node*> node_map = graph.BuildNodeNameIndex(); in VerifyPlacerInspectionRequiredOps() local
|
| 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/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 | freeze_requantization_ranges.cc | 127 std::map<string, const NodeDef*> node_map; in FreezeRequantizationRanges() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 70 std::unordered_map<string, Node*>* node_map) { in BuildNodeMap()
110 const std::unordered_map<string, Node*>& node_map, in GetEngineInfo()
642 std::unordered_map<string, Node*> node_map; in ConvertAfterShapes() local
|