| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_stats.cc | 98 std::unique_ptr<TFGraphNode> node( in TFStats()
99 new TFGraphNode(node_pb.second, profile, &id_to_string_, &nodes_map_)); in TFStats()
100 nodes_map_.insert(std::pair<string, std::unique_ptr<TFGraphNode>>( in TFStats()
204 nodes_map_[node.name()] = std::unique_ptr<TFGraphNode>( in AddGraph()
205 new TFGraphNode(&node, num_nodes, &nodes_map_)); in AddGraph()
209 TFGraphNode* node = nodes_map_.at(it->first).get(); in AddGraph()
299 nodes_map_[name] = std::unique_ptr<TFGraphNode>( in AddRunMeta()
300 new TFGraphNode(&def, num_nodes, &nodes_map_)); in AddRunMeta()
371 void TFStats::AddNodeForTest(int64 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 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 | 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 | 63 const std::map<string, std::unique_ptr<TFGraphNode>>& nodes() const { in nodes()
98 void AddNodeForTest(int64 step, std::unique_ptr<TFGraphNode> node);
115 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 | 38 parent_nodes_[name] = std::unique_ptr<TFGraphNode>( in CreateParentNode()
39 new TFGraphNode(node_defs_.back().get(), -1, nullptr)); in CreateParentNode()
45 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()
394 void TFCode::AddNode(TFGraphNode* node) { in AddNode()
408 std::pair<string, std::vector<TFGraphNode*>>(forward_name, {node})); in AddNode()
447 TFGraphNode* fn = forward_it->second; in Build()
460 for (TFGraphNode* gn : it.second) { in Build()
|
| D | tfprof_show_multi.h | 45 virtual void AddNode(TFGraphNode* node) = 0;
|
| D | tfprof_show.h | 44 virtual void AddNode(TFGraphNode* node) = 0;
|
| D | tfprof_node.cc | 194 void TFGraphNode::AddStepStat(int64 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 | 177 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()
|