Home
last modified time | relevance | path

Searched refs:NodeExecStats (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h35 class NodeExecStats; variable
96 NodeExecStatsWrapper(std::unique_ptr<NodeExecStats> stats,
116 NodeExecStats* stats() { return stats_.get(); } in stats()
128 std::unique_ptr<NodeExecStats> stats_;
170 void Save(const string& device, NodeExecStats* node_stats_pb);
Dstep_stats_collector.cc55 : NodeExecStatsWrapper(MakeUnique<NodeExecStats>(), node, in NodeExecStatsWrapper()
61 std::unique_ptr<NodeExecStats> stats, const NodeDef* node, in NodeExecStatsWrapper()
325 std::unordered_map<string, NodeExecStats> name_to_hw_node_stats; in BuildCostModel()
349 const NodeExecStats& stats = dev_stats.regular_stats->node_stats(i); in BuildCostModel()
372 const NodeExecStats& hw_stats = name_to_hw_node_stats[node_name]; in BuildCostModel()
385 NodeExecStats* node_stats_pb) { in Save()
387 new NodeExecStatsWrapper(std::unique_ptr<NodeExecStats>(node_stats_pb), in Save()
Dprofile_handler.h43 virtual void RecordOneOp(const string& device, const NodeExecStats& stats,
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer_test.cc46 NodeExecStats MakeNodeStats(absl::string_view name, uint32 thread_id, in MakeNodeStats()
48 NodeExecStats ns; in MakeNodeStats()
59 explicit NodeStatsMatcher(const NodeExecStats& expected) in NodeStatsMatcher()
62 bool MatchAndExplain(const NodeExecStats& p, in MatchAndExplain()
75 const NodeExecStats expected_;
79 const NodeExecStats& expected) { in EqualsNodeStats()
Dhost_tracer.cc123 NodeExecStats* ns = dev_stats->add_node_stats(); in CollectData()
/external/tensorflow/tensorflow/python/debug/lib/
Dprofiling_test.py30 node_1 = step_stats_pb2.NodeExecStats(
38 node_2 = step_stats_pb2.NodeExecStats(
46 node_3 = step_stats_pb2.NodeExecStats(
54 node_4 = step_stats_pb2.NodeExecStats(
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py85 node1 = step_stats_pb2.NodeExecStats(
91 node2 = step_stats_pb2.NodeExecStats(
121 node1 = step_stats_pb2.NodeExecStats(
180 node1 = step_stats_pb2.NodeExecStats(
187 node2 = step_stats_pb2.NodeExecStats(
232 node1 = step_stats_pb2.NodeExecStats(
239 node2 = step_stats_pb2.NodeExecStats(
286 node1 = step_stats_pb2.NodeExecStats(
293 node2 = step_stats_pb2.NodeExecStats(
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_cache_logger.cc72 NodeExecStats* ns) { in Save()
102 NodeExecStats* ns = new NodeExecStats; in RecordDataTransfer()
Dworker_cache_logger.h84 void Save(const string& device, int64 step_id, NodeExecStats* ns);
Dmaster_session.cc290 string DetailText(const NodeDetails& details, const NodeExecStats& stats) { in DetailText()
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.h38 class NodeExecStats; variable
107 const NodeExecStats& ns) const;
Dstat_summarizer.cc45 const NodeExecStats& ns) const { in Validate()
90 std::string OpType(const DeviceStepStats& ds, const NodeExecStats& ns) { in OpType()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.cc46 void ExecStep::AddTimeStats(const string& dev, const NodeExecStats& step_stat) { in AddTimeStats()
81 const NodeExecStats& step_stat) { in AddMemoryStats()
195 const NodeExecStats& step_stat) { in AddStepStat()
Dtfprof_node.h90 void AddTimeStats(const string& dev, const NodeExecStats& step_stat);
92 void AddMemoryStats(const string& dev, const NodeExecStats& step_stat);
353 const NodeExecStats& step_stat);
Dtfprof_stats.cc287 for (const NodeExecStats& node_stat : dev_stat.node_stats()) { in AddRunMeta()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto55 message NodeExecStats { message
81 repeated NodeExecStats node_stats = 2;
/external/tensorflow/tensorflow/core/graph/
Dcostmodel_test.cc51 NodeExecStats* node_stats = device_stepstats->add_node_stats(); in GenerateStepStats()
/external/tensorflow/tensorflow/core/grappler/costs/
Dmeasuring_cost_estimator.cc96 for (const NodeExecStats& node_stats : stepstats.node_stats()) { in PredictCosts()
Dvirtual_scheduler.cc1211 NodeExecStats* node_stats = device_stepstats->add_node_stats(); in GenerateRunMetadata()
/external/tensorflow/tensorflow/core/profiler/internal/advisor/
Dtfprof_advisor_test.cc53 NodeExecStats node_stat; in CreateNode()
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler_test.py68 node1 = step_stats_pb2.NodeExecStats(
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_collector.cc337 NodeExecStats* ns = new NodeExecStats; in Flush()
/external/tensorflow/tensorflow/core/protobuf/
Deager_service.proto175 // TODO(nareshmodi): Consider adding NodeExecStats here to be able to
/external/tensorflow/tensorflow/python/kernel_tests/
Dsummary_ops_test.py1001 node_stats=[step_stats_pb2.NodeExecStats(node_name='hello')])