| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_stats.cc | 97 std::unique_ptr<TFGraphNode> node( in TFStats()
98 new TFGraphNode(node_pb.second, profile, &id_to_string_, &nodes_map_)); in TFStats()
99 nodes_map_.insert(std::pair<string, std::unique_ptr<TFGraphNode>>( in TFStats()
203 nodes_map_[node.name()] = std::unique_ptr<TFGraphNode>( in AddGraph()
204 new TFGraphNode(&node, num_nodes, &nodes_map_)); in AddGraph()
208 TFGraphNode* node = nodes_map_.at(it->first).get(); in AddGraph()
298 nodes_map_[name] = std::unique_ptr<TFGraphNode>( in AddRunMeta()
299 new TFGraphNode(&def, num_nodes, &nodes_map_)); in AddRunMeta()
370 void TFStats::AddNodeForTest(int64_t step, std::unique_ptr<TFGraphNode> node) { in AddNodeForTest()
|
| D | tfprof_node.h | 42 class TFGraphNode; variable
307 class TFGraphNode {
309 TFGraphNode(const ProfileNode& node, const ProfileProto& profile, in TFGraphNode() function
311 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map) { in TFGraphNode()
316 TFGraphNode(const NodeDef* node, int64_t id, in TFGraphNode() function
317 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map) { in TFGraphNode()
382 const std::map<string, std::unique_ptr<TFGraphNode>>& nodes_map) { in ToProto()
715 const TFGraphNode* input_node = input_it->second.get(); in input_shapes()
731 const std::map<string, std::unique_ptr<TFGraphNode>>* nodes_map_;
793 std::vector<const TFGraphNode*> nodes = pick_nodes(type_regexes); in SnapshotNodes()
[all …]
|
| D | tfprof_node_show.h | 43 explicit ShowNode(const TFGraphNode* node);
58 const TFGraphNode* node;
68 explicit GraphNode(TFGraphNode* node) : ShowNode(node) {} in GraphNode()
78 explicit ScopeNode(const TFGraphNode* node) : ShowNode(node) {} in ScopeNode()
|
| D | tfprof_code.h | 59 void AddNode(TFGraphNode* node) override;
90 std::map<string, std::vector<TFGraphNode*>> grad_nodes_;
91 std::map<string, TFGraphNode*> forward_nodes_;
|
| D | tfprof_stats.h | 63 const std::map<string, std::unique_ptr<TFGraphNode>>& nodes() const { in nodes()
98 void AddNodeForTest(int64_t step, std::unique_ptr<TFGraphNode> node);
115 std::map<string, std::unique_ptr<TFGraphNode>> nodes_map_;
|
| D | tfprof_scope.h | 45 void AddNode(TFGraphNode* node) override;
70 std::map<string, std::unique_ptr<TFGraphNode>> parent_nodes_;
|
| D | tfprof_graph.h | 47 void AddNode(TFGraphNode* node) override;
80 std::map<string, std::unique_ptr<TFGraphNode>> parent_nodes_;
|
| D | tfprof_scope.cc | 36 parent_nodes_[name] = std::unique_ptr<TFGraphNode>( in CreateParentNode()
37 new TFGraphNode(node_defs_.back().get(), -1, nullptr)); in CreateParentNode()
43 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 | 45 void AddNode(TFGraphNode* node) override;
|
| D | tfprof_code.cc | 191 const TFGraphNode* gn = gn_it.second; in Add()
392 void TFCode::AddNode(TFGraphNode* node) { in AddNode()
406 std::pair<string, std::vector<TFGraphNode*>>(forward_name, {node})); in AddNode()
445 TFGraphNode* fn = forward_it->second; in Build()
458 for (TFGraphNode* gn : it.second) { in Build()
|
| D | tfprof_show_multi.h | 43 virtual void AddNode(TFGraphNode* node) = 0;
|
| D | tfprof_show.h | 43 virtual void AddNode(TFGraphNode* node) = 0;
|
| D | tfprof_node.cc | 191 void TFGraphNode::AddStepStat(int64_t step, const string& device, in AddStepStat()
|
| D | tfprof_node_show.cc | 21 ShowNode::ShowNode(const TFGraphNode* node) : node(node), account(false) { in ShowNode()
|
| D | tfprof_op.cc | 81 void TFOp::AddNode(TFGraphNode* node) { in AddNode()
|
| D | tfprof_timeline.cc | 180 const TFGraphNode* node = gnode->node; in AllocateTimeNodes()
|
| /external/tensorflow/tensorflow/core/profiler/internal/advisor/ |
| D | tfprof_advisor_test.cc | 38 std::unique_ptr<TFGraphNode> CreateNode(const string& name, in CreateNode()
51 std::unique_ptr<TFGraphNode> node(new TFGraphNode(def, -1, nullptr)); in CreateNode()
|
| D | operation_checker.h | 42 const TFGraphNode* node = n.second.get(); in Check()
|
| D | accelerator_utilization_checker.h | 73 void BuildExecStats(const TFGraphNode* node) { in BuildExecStats()
|