Searched refs:NodeExecStatsWrapper (Results 1 – 2 of 2) sorted by relevance
53 NodeExecStatsWrapper::NodeExecStatsWrapper( in NodeExecStatsWrapper() function in tensorflow::NodeExecStatsWrapper55 : NodeExecStatsWrapper(MakeUnique<NodeExecStats>(), node, in NodeExecStatsWrapper()60 NodeExecStatsWrapper::NodeExecStatsWrapper( in NodeExecStatsWrapper() function in tensorflow::NodeExecStatsWrapper67 void NodeExecStatsWrapper::Done(const string& device) { in Done()111 void NodeExecStatsWrapper::RecordExecutorStarted() { in RecordExecutorStarted()117 void NodeExecStatsWrapper::RecordComputeStarted() { in RecordComputeStarted()126 void NodeExecStatsWrapper::RecordComputeEnded() { in RecordComputeEnded()135 void NodeExecStatsWrapper::RecordExecutorEnded() { in RecordExecutorEnded()144 void NodeExecStatsWrapper::SetScheduled(int64 nanos) { in SetScheduled()149 void NodeExecStatsWrapper::SetMemory(OpKernelContext* ctx) { in SetMemory()[all …]
89 class NodeExecStatsWrapper : public NodeExecStatsInterface {92 NodeExecStatsWrapper(const NodeDef* node,96 NodeExecStatsWrapper(std::unique_ptr<NodeExecStats> stats,101 ~NodeExecStatsWrapper() override { Finalize(); } in ~NodeExecStatsWrapper()171 void Save(const string& device, NodeExecStatsWrapper* node_stats);192 typedef std::vector<std::unique_ptr<NodeExecStatsWrapper>> NodeStatsVector;