HomeSort by relevance Sort by last modified time
    Searched refs:NodeExecStatsWrapper (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
step_stats_collector.h 38 class NodeExecStatsWrapper {
40 NodeExecStatsWrapper();
42 NodeExecStatsWrapper(NodeExecStats* stats);
45 ~NodeExecStatsWrapper() { Finalize(); }
83 void Save(const string& device, NodeExecStatsWrapper* stats);
102 typedef std::vector<std::unique_ptr<NodeExecStatsWrapper>> NodeExecStatsVec;
step_stats_collector.cc 39 NodeExecStatsWrapper::NodeExecStatsWrapper()
40 : NodeExecStatsWrapper(new NodeExecStats) {}
41 NodeExecStatsWrapper::NodeExecStatsWrapper(NodeExecStats* stats)
44 void NodeExecStatsWrapper::AddAllocation(
59 void NodeExecStatsWrapper::Finalize() {
259 Save(device, new NodeExecStatsWrapper(nt));
263 NodeExecStatsWrapper* stats) {
277 dss.push_back(std::unique_ptr<NodeExecStatsWrapper>(stats))
    [all...]
executor.cc 77 bool SetTimelineLabel(const Node* node, NodeExecStatsWrapper* stats) {
129 void SetScheduled(NodeExecStatsWrapper* stats, int64 t) {
134 void SetAllStart(NodeExecStatsWrapper* stats) {
139 void SetOpStart(NodeExecStatsWrapper* stats) {
146 void SetOpEnd(NodeExecStatsWrapper* stats) {
153 void SetAllEnd(NodeExecStatsWrapper* stats) {
160 void SetOutput(NodeExecStatsWrapper* stats, int slot, const Tensor* v) {
168 void SetMemory(NodeExecStatsWrapper* stats, OpKernelContext* ctx) {
182 void SetReferencedTensors(NodeExecStatsWrapper* stats,
    [all...]

Completed in 135 milliseconds