Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h37 class StepStats; variable
159 explicit StepStatsCollector(StepStats* step_stats);
185 void FinalizeAndSwap(StepStats* step_stats);
201 StepStats* step_stats_ TF_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.h31 class StepStats; variable
85 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;
Dmaster_session.h168 std::vector<StepStats> step_stats; // per partition
169 StepStats rpc_stats; // for RPC layer
Dmessage_wrappers.h491 virtual StepStats* mutable_step_stats() = 0;
526 StepStats* mutable_step_stats() override;
542 StepStats step_stats_;
559 StepStats* mutable_step_stats() override;
586 StepStats* mutable_step_stats() override;
Dworker_cache_wrapper.h87 bool RetrieveLogs(int64 step_id, StepStats* ss) override { in RetrieveLogs()
Dworker_cache_logger.cc59 bool WorkerCacheLogger::RetrieveLogs(int64 step_id, StepStats* ss) { in RetrieveLogs()
Dsession_mgr.cc320 auto step_stats = StepStats(); in RetrieveLogs()
333 auto step_stats = StepStats(); in RetrieveLogs()
Dworker_session.cc87 bool RetrieveLogs(int64 step_id, StepStats* ss) override { in RetrieveLogs()
Dmessage_wrappers.cc635 StepStats* InMemoryRunGraphResponse::mutable_step_stats() { in mutable_step_stats()
703 StepStats* OwnedProtoRunGraphResponse::mutable_step_stats() { in mutable_step_stats()
774 StepStats* NonOwnedProtoRunGraphResponse::mutable_step_stats() { in mutable_step_stats()
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.h37 class StepStats; variable
58 void ProcessStepStats(const StepStats& step_stats);
Dstat_summarizer.cc130 void StatSummarizer::ProcessStepStats(const StepStats& step_stats) { in ProcessStepStats()
/external/tensorflow/tensorflow/core/graph/
Dcostmodel_test.cc45 static void GenerateStepStats(Graph* graph, StepStats* step_stats, in GenerateStepStats()
80 StepStats step_stats; in TEST()
Dcostmodel.h36 class StepStats; variable
88 void MergeFromStats(const NodeNameToCostIdMap& map, const StepStats& ss);
/external/tensorflow/tensorflow/python/util/
Dstat_summarizer_wrapper.cc45 tensorflow::StepStats step_stats; in PYBIND11_MODULE()
/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/grappler/
Dcluster.py101 step_stats_pb2.StepStats.FromString(step_stats_bytes))
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer.cc111 StepStats* step_stats = run_metadata->mutable_step_stats(); in CollectData()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto86 message StepStats { message
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_cache.cc87 bool RetrieveLogs(int64 step_id, StepStats* ss) override { in RetrieveLogs()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_collector.h239 virtual void Export(StepStats* step_stats) {} in Export()
Ddevice_tracer.cc205 StepStats step_stats; in CollectData()

12