| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_stats.cc | 96 std::unique_ptr<TFGraphNode> node( in TFStats()
97 new TFGraphNode(node_pb.second, profile, &id_to_string_, &nodes_map_)); in TFStats()
98 nodes_map_.insert(std::pair<string, std::unique_ptr<TFGraphNode>>( in TFStats()
201 nodes_map_[node.name()] = std::unique_ptr<TFGraphNode>( in AddGraph()
202 new TFGraphNode(&node, nodes_map_.size(), &nodes_map_)); in AddGraph()
206 TFGraphNode* node = nodes_map_.at(it->first).get(); in AddGraph()
295 nodes_map_[name] = std::unique_ptr<TFGraphNode>( in AddRunMeta()
296 new TFGraphNode(&def, nodes_map_.size(), &nodes_map_)); in AddRunMeta()
367 void TFStats::AddNodeForTest(int64 step, std::unique_ptr<TFGraphNode> node) { in AddNodeForTest()
|
| D | tfprof_node.h | 43 class TFGraphNode; variable
308 class TFGraphNode {
310 TFGraphNode(const ProfileNode& node, const ProfileProto& profile, in TFGraphNode() function
312 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map) { in TFGraphNode()
317 TFGraphNode(const NodeDef* node, int64 id, in TFGraphNode() function
318 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map) { in TFGraphNode()
383 const std::map<string, std::unique_ptr<TFGraphNode>>& nodes_map) { in ToProto()
716 const TFGraphNode* input_node = input_it->second.get(); in input_shapes()
732 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map_;
794 std::vector<const TFGraphNode*> nodes = pick_nodes(type_regexes); in SnapshotNodes()
[all …]
|
| D | tfprof_node_show.h | 44 explicit ShowNode(const TFGraphNode* node);
59 const TFGraphNode* node;
69 explicit GraphNode(TFGraphNode* node) : ShowNode(node) {} in GraphNode()
79 explicit ScopeNode(const TFGraphNode* node) : ShowNode(node) {} in ScopeNode()
|
| D | tfprof_code.h | 61 void AddNode(TFGraphNode* node) override;
91 std::map<string, std::vector<TFGraphNode*>> grad_nodes_;
92 std::map<string, TFGraphNode*> forward_nodes_;
|
| D | tfprof_stats.h | 64 const std::map<string, std::unique_ptr<TFGraphNode>>& nodes() const { in nodes()
99 void AddNodeForTest(int64 step, std::unique_ptr<TFGraphNode> node);
116 std::map<string, std::unique_ptr<TFGraphNode>> nodes_map_;
|
| D | tfprof_scope.h | 46 void AddNode(TFGraphNode* node) override;
71 std::map<string, std::unique_ptr<TFGraphNode>> parent_nodes_;
|
| D | tfprof_graph.h | 49 void AddNode(TFGraphNode* node) override;
82 std::map<string, std::unique_ptr<TFGraphNode>> parent_nodes_;
|
| D | tfprof_scope.cc | 37 parent_nodes_[name] = std::unique_ptr<TFGraphNode>( in CreateParentNode()
38 new TFGraphNode(node_defs_.back().get(), -1, nullptr)); in CreateParentNode()
44 void TFScope::AddNode(TFGraphNode* node) { in AddNode()
|
| D | tfprof_graph.cc | 33 parent_nodes_[name] = std::unique_ptr<TFGraphNode>( in CreateParentNode()
34 new TFGraphNode(node_defs_.back().get(), -1, nullptr)); in CreateParentNode()
40 void TFGraph::AddNode(TFGraphNode* node) { in AddNode()
|
| D | tfprof_op.h | 47 void AddNode(TFGraphNode* node) override;
|
| D | tfprof_code.cc | 193 const TFGraphNode* gn = gn_it.second; in Add()
383 void TFCode::AddNode(TFGraphNode* node) { in AddNode()
397 std::pair<string, std::vector<TFGraphNode*>>(forward_name, {node})); in AddNode()
435 TFGraphNode* fn = forward_it->second; in Build()
447 for (TFGraphNode* gn : it.second) { in Build()
|
| D | tfprof_show_multi.h | 46 virtual void AddNode(TFGraphNode* node) = 0;
|
| D | tfprof_show.h | 45 virtual void AddNode(TFGraphNode* node) = 0;
|
| D | tfprof_node.cc | 195 void TFGraphNode::AddStepStat(int64 step, const string& device, in AddStepStat()
|
| D | tfprof_node_show.cc | 24 ShowNode::ShowNode(const TFGraphNode* node) : node(node), account(false) { in ShowNode()
|
| D | tfprof_op.cc | 84 void TFOp::AddNode(TFGraphNode* node) { in AddNode()
|
| D | tfprof_timeline.cc | 177 const TFGraphNode* node = gnode->node; in AllocateTimeNodes()
|
| /external/tensorflow/tensorflow/core/profiler/internal/advisor/ |
| D | tfprof_advisor_test.cc | 39 std::unique_ptr<TFGraphNode> CreateNode(const string& name, in CreateNode()
52 std::unique_ptr<TFGraphNode> node(new TFGraphNode(def, -1, nullptr)); in CreateNode()
|
| D | operation_checker.h | 41 const TFGraphNode* node = n.second.get(); in Check()
|
| D | accelerator_utilization_checker.h | 72 void BuildExecStats(const TFGraphNode* node) { in BuildExecStats()
|