Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer_test.cc37 NodeExecStats MakeNodeStats(const string& name, uint64 thread_id, in MakeNodeStats()
39 NodeExecStats ns; in MakeNodeStats()
50 explicit NodeStatsMatcher(const NodeExecStats& expected) in NodeStatsMatcher()
53 bool MatchAndExplain(const NodeExecStats& p, in MatchAndExplain()
66 const NodeExecStats expected_;
70 const NodeExecStats& expected) { in EqualsNodeStats()
Dhost_tracer.cc90 NodeExecStats* ns = new NodeExecStats; in CollectDataToCollector()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h39 class NodeExecStats; variable
82 std::vector<Tensor>* outputs, NodeExecStats* stats,
88 std::vector<Tensor>* outputs, NodeExecStats* stats,
148 std::vector<Tensor>* outputs, NodeExecStats* stats,
153 std::vector<Tensor>* outputs, NodeExecStats* stats,
210 std::vector<Tensor>* outputs, NodeExecStats* stats,
214 std::vector<Tensor>* outputs, NodeExecStats* stats,
Dexecute_node.h36 KernelAndDevice* kernel, NodeExecStats* maybe_stats, in ExecuteNode()
83 std::unique_ptr<NodeExecStats> maybe_stats_;
Dkernel_and_device.cc153 NodeExecStats* stats, StepStats* step_stats, in Run()
164 std::vector<Tensor>* outputs, NodeExecStats* stats, StepStats* step_stats, in Run()
179 NodeExecStats* stats) { in UpdateStats()
209 NodeExecStats* stats, StepStats* step_stats, in Run()
307 std::vector<Tensor>* outputs, NodeExecStats* stats, StepStats* step_stats, in Run()
Dexecute.h48 KernelAndDevice* kernel, NodeExecStats* maybe_stats,
Dexecute.cc479 std::unique_ptr<NodeExecStats> maybe_stats; in EagerLocalExecute()
488 maybe_stats.reset(new NodeExecStats); in EagerLocalExecute()
914 KernelAndDevice* kernel, NodeExecStats* maybe_stats, in EagerKernelExecute()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h36 class NodeExecStats; variable
101 NodeExecStatsWrapper(std::unique_ptr<NodeExecStats> stats, const Node* node,
121 NodeExecStats* stats() { return stats_.get(); } in stats()
133 std::unique_ptr<NodeExecStats> stats_;
175 void Save(const string& device, NodeExecStats* node_stats_pb);
Dstep_stats_collector.cc46 : NodeExecStatsWrapper(MakeUnique<NodeExecStats>(), node, in NodeExecStatsWrapper()
52 std::unique_ptr<NodeExecStats> stats, const Node* node, in NodeExecStatsWrapper()
327 std::unordered_map<string, NodeExecStats> name_to_hw_node_stats; in BuildCostModel()
351 const NodeExecStats& stats = dev_stats.regular_stats->node_stats(i); in BuildCostModel()
374 const NodeExecStats& hw_stats = name_to_hw_node_stats[node_name]; in BuildCostModel()
387 NodeExecStats* node_stats_pb) { in Save()
389 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/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);
/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.cc47 void ExecStep::AddTimeStats(const string& dev, const NodeExecStats& step_stat) { in AddTimeStats()
82 const NodeExecStats& step_stat) { in AddMemoryStats()
196 const NodeExecStats& step_stat) { in AddStepStat()
Dtfprof_node.h91 void AddTimeStats(const string& dev, const NodeExecStats& step_stat);
93 void AddMemoryStats(const string& dev, const NodeExecStats& step_stat);
354 const NodeExecStats& step_stat);
Dtfprof_stats.cc284 for (const NodeExecStats& node_stat : dev_stat.node_stats()) { in AddRunMeta()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto53 message NodeExecStats { message
79 repeated NodeExecStats node_stats = 2;
/external/tensorflow/tensorflow/core/platform/default/
Ddevice_tracer.cc577 Status SaveStats(std::unique_ptr<NodeExecStats> stats, in SaveStats()
585 new NodeExecStats(*stats)); in SaveStats()
610 auto stats = absl::make_unique<NodeExecStats>(); in SaveRecord()
635 auto stats = absl::make_unique<NodeExecStats>(); in SaveRecord()
/external/tensorflow/tensorflow/core/grappler/costs/
Dmeasuring_cost_estimator.cc96 for (const NodeExecStats& node_stats : stepstats.node_stats()) { in PredictCosts()
/external/tensorflow/tensorflow/core/profiler/internal/advisor/
Dtfprof_advisor_test.cc54 NodeExecStats node_stat; in CreateNode()
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler_test.py68 node1 = step_stats_pb2.NodeExecStats(
/external/tensorflow/tensorflow/core/protobuf/
Deager_service.proto105 // TODO(nareshmodi): Consider adding NodeExecStats here to be able to

12