Home
last modified time | relevance | path

Searched refs:StepStats (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h40 class StepStats; variable
83 StepStats* step_stats,
89 StepStats* step_stats,
149 StepStats* step_stats, GraphCollector* graph_collector) override;
154 StepStats* step_stats, GraphCollector* graph_collector) override;
211 StepStats* step_stats, GraphCollector* graph_collector) override;
215 StepStats* step_stats, GraphCollector* graph_collector) override;
Dexecute_node.h37 StepStats* maybe_step_stats, GraphCollector* graph_collector, in ExecuteNode()
84 StepStats* maybe_step_stats_;
Dkernel_and_device.cc153 NodeExecStats* stats, StepStats* step_stats, in Run()
164 std::vector<Tensor>* outputs, NodeExecStats* stats, StepStats* step_stats, in Run()
209 NodeExecStats* stats, StepStats* step_stats, in Run()
307 std::vector<Tensor>* outputs, NodeExecStats* stats, StepStats* step_stats, in Run()
Dexecute.h49 StepStats* maybe_step_stats,
Dexecute.cc57 void MaybeInitializeStepStats(StepStats* step_stats, EagerContext* ctx) { in MaybeInitializeStepStats()
67 int StepStatsDeviceIndex(StepStats* step_stats, EagerContext* ctx, in StepStatsDeviceIndex()
480 StepStats* maybe_step_stats = nullptr; in EagerLocalExecute()
915 StepStats* maybe_step_stats, in EagerKernelExecute()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h38 class StepStats; variable
164 explicit StepStatsCollector(StepStats* step_stats);
190 void FinalizeAndSwap(StepStats* step_stats);
206 StepStats* step_stats_ GUARDED_BY(mu_);
Dstats_publisher_interface.cc29 void PublishStatsProto(const StepStats& step_stats) override {} in PublishStatsProto()
Dstats_publisher_interface.h35 virtual void PublishStatsProto(const StepStats& step_stats) = 0;
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_cache.h30 class StepStats; variable
83 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) { return false; } in RetrieveLogs()
Dworker_cache_logger.h48 bool RetrieveLogs(int64 step_id, StepStats* ss);
76 StepStats step_stats;
Dmessage_wrappers.h479 virtual StepStats* mutable_step_stats() = 0;
514 StepStats* mutable_step_stats() override;
530 StepStats step_stats_;
547 StepStats* mutable_step_stats() override;
574 StepStats* mutable_step_stats() override;
Dsession_mgr.cc174 auto step_stats = StepStats(); in RetrieveLogs()
187 auto step_stats = StepStats(); in RetrieveLogs()
Dmaster_session.h170 std::vector<StepStats> step_stats; // per partition
171 StepStats rpc_stats; // for RPC layer
Dworker_cache_wrapper.h86 virtual bool RetrieveLogs(int64 step_id, StepStats* ss) { in RetrieveLogs()
Dworker_cache_logger.cc59 bool WorkerCacheLogger::RetrieveLogs(int64 step_id, StepStats* ss) { in RetrieveLogs()
Dworker_session.cc75 bool RetrieveLogs(int64 step_id, StepStats* ss) override { in RetrieveLogs()
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.h37 class StepStats; variable
58 void ProcessStepStats(const StepStats& step_stats);
/external/tensorflow/tensorflow/core/platform/
Ddevice_tracer_test.cc118 StepStats stats; in TEST_F()
128 StepStats stats; in TEST_F()
196 StepStats stats; in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dcostmodel.h36 class StepStats; variable
80 void MergeFromStats(const NodeNameToCostIdMap& map, const StepStats& ss);
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-run-metadata.pbtxt10 type_name: ".tensorflow.StepStats"
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory.h70 void InferFromTrace(const StepStats& timeline);
/external/tensorflow/tensorflow/python/util/
Dstat_summarizer.i63 tensorflow::StepStats step_stats; in ProcessStepStatsStr()
/external/tensorflow/tensorflow/python/grappler/
Dcluster.py108 step_stats_pb2.StepStats.FromString(step_stats_bytes))
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_cache.cc97 bool RetrieveLogs(int64 step_id, StepStats* ss) override { in RetrieveLogs()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto84 message StepStats { message

12