Searched refs:NodeExecStatsWrapper (Results 1 – 2 of 2) sorted by relevance
44 NodeExecStatsWrapper::NodeExecStatsWrapper( in NodeExecStatsWrapper() function in tensorflow::NodeExecStatsWrapper46 : NodeExecStatsWrapper(MakeUnique<NodeExecStats>(), node, in NodeExecStatsWrapper()51 NodeExecStatsWrapper::NodeExecStatsWrapper( in NodeExecStatsWrapper() function in tensorflow::NodeExecStatsWrapper58 void NodeExecStatsWrapper::Done(const string& device) { in Done()103 void NodeExecStatsWrapper::RecordExecutorStarted() { in RecordExecutorStarted()109 void NodeExecStatsWrapper::RecordComputeStarted() { in RecordComputeStarted()118 void NodeExecStatsWrapper::RecordComputeEnded() { in RecordComputeEnded()127 void NodeExecStatsWrapper::RecordExecutorEnded() { in RecordExecutorEnded()136 void NodeExecStatsWrapper::SetScheduled(int64 nanos) { in SetScheduled()141 void NodeExecStatsWrapper::SetMemory(OpKernelContext* ctx) { in SetMemory()[all …]
94 class NodeExecStatsWrapper : public NodeExecStatsInterface {97 NodeExecStatsWrapper(const Node* node,101 NodeExecStatsWrapper(std::unique_ptr<NodeExecStats> stats, const Node* node,105 ~NodeExecStatsWrapper() { Finalize(); } in ~NodeExecStatsWrapper()176 void Save(const string& device, NodeExecStatsWrapper* node_stats);197 typedef std::vector<std::unique_ptr<NodeExecStatsWrapper>> NodeStatsVector;