Home
last modified time | relevance | path

Searched defs:step_stats (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dcostmodel_test.cc45 static void GenerateStepStats(Graph* graph, StepStats* step_stats, in GenerateStepStats()
80 StepStats step_stats; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dsession_mgr.cc368 auto step_stats = StepStats(); in RetrieveLogs() local
381 auto step_stats = StepStats(); in RetrieveLogs() local
H A Dworker_cache_logger.h76 StepStats step_stats; member
H A Dmaster_session.h168 std::vector<StepStats> step_stats; // per partition member
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dworker.proto296 StepStats step_stats = 2; field
430 StepStats step_stats = 2; field
H A Dconfig.proto796 StepStats step_stats = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dstep_stats_collector.cc196 StepStatsCollector::StepStatsCollector(StepStats* step_stats) in StepStatsCollector()
519 void StepStatsCollector::FinalizeAndSwap(StepStats* step_stats) { in FinalizeAndSwap()
H A Dstats_publisher_interface.cc29 void PublishStatsProto(const StepStats& step_stats) override {} in PublishStatsProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/lib/
H A Ddevice_profiler_session.h52 Status CollectData(StepStats* step_stats) { in CollectData()
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dstat_summarizer_wrapper.cc45 tensorflow::StepStats step_stats; in PYBIND11_MODULE() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dstat_summarizer.cc130 void StatSummarizer::ProcessStepStats(const StepStats& step_stats) { in ProcessStepStats()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/
H A Dxplane_to_step_stats.cc71 void ConvertGpuXSpaceToStepStats(const XSpace& xspace, StepStats* step_stats) { in ConvertGpuXSpaceToStepStats()
/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dcluster_wrapper.cc271 tensorflow::StepStats step_stats; in PYBIND11_MODULE() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dutils.cc451 const auto& step_stats = run_metadata.step_stats(); in GetStatsStringFromRunMetadata() local
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dtimeline.py346 def __init__(self, step_stats, graph=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/
H A Dbenchmark_model.cc318 const StepStats& step_stats = run_metadata.step_stats(); in RunBenchmark() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dworker.pb.h7905 inline const ::tensorflow::StepStats& RunGraphResponse::step_stats() const { in step_stats() function
7910 ::tensorflow::StepStats* step_stats) { in unsafe_arena_set_allocated_step_stats()
7952 inline void RunGraphResponse::set_allocated_step_stats(::tensorflow::StepStats* step_stats) { in set_allocated_step_stats()
8931 inline const ::tensorflow::StepStats& LabeledStepStats::step_stats() const { in step_stats() function
8936 ::tensorflow::StepStats* step_stats) { in unsafe_arena_set_allocated_step_stats()
8978 inline void LabeledStepStats::set_allocated_step_stats(::tensorflow::StepStats* step_stats) { in set_allocated_step_stats()
Dconfig.pb.h7636 inline const ::tensorflow::StepStats& RunMetadata::step_stats() const { in step_stats() function
7641 ::tensorflow::StepStats* step_stats) { in unsafe_arena_set_allocated_step_stats()
7683 inline void RunMetadata::set_allocated_step_stats(::tensorflow::StepStats* step_stats) { in set_allocated_step_stats()
Dworker.pb.cc4079 RunGraphResponse::_Internal::step_stats(const RunGraphResponse* msg) { in step_stats() function in tensorflow::RunGraphResponse::_Internal
6198 LabeledStepStats::_Internal::step_stats(const LabeledStepStats* msg) { in step_stats() function in tensorflow::LabeledStepStats::_Internal
Dconfig.pb.cc6102 RunMetadata::_Internal::step_stats(const RunMetadata* msg) { in step_stats() function in tensorflow::RunMetadata::_Internal
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dbasic_session_run_hooks.py1088 def _save(self, step, save_path, step_stats): argument