Home
last modified time | relevance | path

Searched refs:step_num (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/convert/
Dstep_events_to_steps_db.cc40 StepInfoResult ConvertStepDetailsToStepInfo(bool has_device, int64 step_num, in ConvertStepDetailsToStepInfo() argument
64 step_info.set_step_num(step_num); in ConvertStepDetailsToStepInfo()
96 out << "step_num=" << step_info.step_num() in DebugStepInfo()
Dop_stats_to_pod_viewer_test.cc89 EXPECT_EQ(kStepNum, pod_stats_map.step_num()); in TEST()
91 EXPECT_EQ(kStepNum, record.step_num()); in TEST()
Dop_stats_to_pod_viewer.cc36 pod_stats_map->set_step_num(step_sequence.step_num()); in ConvertOpStatsToPodStatsSequence()
Dop_stats_to_pod_stats.cc39 record.set_step_num(step_info.step_num()); in CreatePodStatsRecord()
Dop_stats_to_pod_stats_test.cc87 EXPECT_EQ(kStepNum, record.step_num()); in TEST()
Dop_stats_combiner.cc49 uint32 new_step_num = dst->step_num(); in CombinePerCoreStepInfo()
Dop_stats_to_input_pipeline_analysis.cc212 details.set_step_number(step_info.step_num()); in ComputeGenericInputPipelineAnalysisResult()
/external/tensorflow/tensorflow/core/profiler/utils/
Dstep_intersection_test.cc36 uint32 step_num = in CreateOneTestStep() local
39 result.set_step_num(step_num); in CreateOneTestStep()
41 info.set_step_num(step_num); in CreateOneTestStep()
Dgroup_events.cc417 int64 step_num = group_id_.value_or(0); in GetGroupName() local
419 step_num = stat->IntValue(); in GetGroupName()
422 step_num = stat->IntValue(); in GetGroupName()
424 absl::StrAppend(&name, step_num); in GetGroupName()
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dpod_stats.proto33 uint32 step_num = 4; field
Dsteps_db.proto29 uint32 step_num = 1; field
90 uint32 step_num = 1; field
Dpod_viewer.proto37 uint32 step_num = 1; field
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py1299 def _inspect_summary_cache(self, cache, replica_id, step_num, output_stream, argument
1342 stats = ['\n\n', 'core:', replica_id, ',', 'step:', step_num, '-->',
1346 stats = ['\n\n', 'step:', step_num, '-->', step_error_message,
1354 '\n', 'core:', replica_id, ',', 'step:', step_num, ',',
1358 '\n', 'step:', step_num, ',',
1381 def _flush_fun(cache, replica_id, step_num): argument
1403 cache=cache, replica_id=replica_id, step_num=step_num,
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py1148 step_num=step,
1444 with trace.Trace('test', step_num=step, _r=1):