Home
last modified time | relevance | path

Searched refs:host_plane (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_trace_events_test.cc30 XPlaneBuilder host_plane(space->add_planes()); in CreateXSpace() local
31 host_plane.SetName(kHostThreadsPlaneName); in CreateXSpace()
32 XLineBuilder thread1 = host_plane.GetOrCreateLine(10); in CreateXSpace()
35 thread1.AddEvent(*host_plane.GetOrCreateEventMetadata("event1")); in CreateXSpace()
38 event1.AddStatValue(*host_plane.GetOrCreateStatMetadata("tf_op"), in CreateXSpace()
39 *host_plane.GetOrCreateStatMetadata("Relu")); in CreateXSpace()
40 XLineBuilder thread2 = host_plane.GetOrCreateLine(20); in CreateXSpace()
43 thread2.AddEvent(*host_plane.GetOrCreateEventMetadata("event2")); in CreateXSpace()
46 event2.AddStatValue(*host_plane.GetOrCreateStatMetadata("tf_op"), in CreateXSpace()
47 *host_plane.GetOrCreateStatMetadata("Conv2D")); in CreateXSpace()
Dxplane_to_profile_response_test.cc31 XPlaneBuilder host_plane(space->add_planes()); in CreateXSpace() local
34 host_plane.SetName("cpu"); in CreateXSpace()
35 host_plane.SetId(0); in CreateXSpace()
36 XLineBuilder thread1 = host_plane.GetOrCreateLine(10); in CreateXSpace()
39 thread1.AddEvent(*host_plane.GetOrCreateEventMetadata("event1")); in CreateXSpace()
42 event1.AddStatValue(*host_plane.GetOrCreateStatMetadata("tf_op"), in CreateXSpace()
43 *host_plane.GetOrCreateStatMetadata("Relu")); in CreateXSpace()
44 XLineBuilder thread2 = host_plane.GetOrCreateLine(20); in CreateXSpace()
47 thread2.AddEvent(*host_plane.GetOrCreateEventMetadata("event2")); in CreateXSpace()
50 event2.AddStatValue(*host_plane.GetOrCreateStatMetadata("tf_op"), in CreateXSpace()
[all …]
Dpost_process_single_host_xplane.cc28 XPlane* host_plane = in MergeHostPlanesAndSortLines() local
33 MergePlanes(additional_host_planes, host_plane); in MergeHostPlanesAndSortLines()
36 SortXLinesBy(host_plane, XLinesComparatorByName()); in MergeHostPlanesAndSortLines()
Dxplane_to_op_metrics_db_test.cc58 XPlaneBuilder host_plane(xplane); in TEST() local
59 XLineBuilder thread1 = host_plane.GetOrCreateLine(/*line_id=*/10); in TEST()
62 /*kernel_name=*/"", &host_plane, &thread1); in TEST()
63 XLineBuilder thread2 = host_plane.GetOrCreateLine(/*line_id=*/20); in TEST()
66 /*kernel_name=*/"", &host_plane, &thread2); in TEST()
69 /*kernel_name=*/"", &host_plane, &thread2); in TEST()
Dxplane_to_op_stats.cc112 void ProcessHostPlane(const XPlane* host_plane, bool use_device_step_events, in ProcessHostPlane() argument
116 CollectTfOpsFromHostThreadsXPlane(*host_plane); in ProcessHostPlane()
118 XPlaneVisitor plane = CreateTfXPlaneVisitor(host_plane); in ProcessHostPlane()
150 const XPlane* host_plane = FindPlaneWithName(space, kHostThreadsPlaneName); in ConvertXSpaceToOpStats() local
201 if (host_plane && options.generate_op_metrics_db) { in ConvertXSpaceToOpStats()
202 ProcessHostPlane(host_plane, has_device, options, in ConvertXSpaceToOpStats()
Dxplane_to_step_events_test.cc48 XPlane* host_plane = GetOrCreateHostXPlane(&space); in TEST() local
49 XPlaneBuilder host_plane_builder(host_plane); in TEST()
88 *host_plane, true, device_step_events); in TEST()
Dxplane_to_memory_profile_test.cc36 XPlane* host_plane = GetOrCreateHostXPlane(&space); in TEST() local
37 XPlaneBuilder host_plane_builder(host_plane); in TEST()
89 MemoryProfile memory_profile = ConvertXPlaneToMemoryProfile(*host_plane); in TEST()
Dxplane_to_trace_events.cc124 const XPlane* host_plane = FindPlaneWithName(xspace, kHostThreadsPlaneName); in ConvertXSpaceToTraceEvents() local
125 if (host_plane != nullptr) { in ConvertXSpaceToTraceEvents()
126 XPlaneVisitor xplane = CreateTfXPlaneVisitor(host_plane); in ConvertXSpaceToTraceEvents()
Dxplane_to_tf_data_stats_test.cc42 XPlane host_plane; in TEST() local
43 XPlaneBuilder host_plane_builder(&host_plane); in TEST()
78 builder.Add("host1", &host_plane); in TEST()
Dxplane_to_tf_functions_test.cc46 const XPlane* host_plane = FindPlaneWithName(space, kHostThreadsPlaneName); in ConvertXSpaceToTfFunctionDb() local
47 if (host_plane) { in ConvertXSpaceToTfFunctionDb()
48 XPlaneVisitor plane = CreateTfXPlaneVisitor(host_plane); in ConvertXSpaceToTfFunctionDb()
Dxplane_to_tools_data.cc202 XPlane* host_plane = in ConvertMultiXSpacesToTfDataBottleneckAnalysis() local
204 if (host_plane == nullptr) { in ConvertMultiXSpacesToTfDataBottleneckAnalysis()
211 builder.Add(host_name, host_plane); in ConvertMultiXSpacesToTfDataBottleneckAnalysis()
Dxplane_to_memory_profile.h30 MemoryProfile ConvertXPlaneToMemoryProfile(const XPlane& host_plane,
Dxplane_to_tf_data_stats.h49 void Add(absl::string_view host_name, XPlane* host_plane);
Dxplane_to_memory_profile.cc536 MemoryProfile ConvertXPlaneToMemoryProfile(const XPlane& host_plane, in ConvertXPlaneToMemoryProfile() argument
538 MemoryProfile memory_profile = GenerateMemoryProfile(&host_plane); in ConvertXPlaneToMemoryProfile()
545 if (const XPlane* host_plane = in ConvertXSpaceToMemoryProfileJson() local
547 MemoryProfile memory_profile = ConvertXPlaneToMemoryProfile(*host_plane); in ConvertXSpaceToMemoryProfileJson()
Dxplane_to_tf_data_stats.cc486 XPlane* host_plane) { in Add() argument
491 event_forest.AddPlanes(CreateTfXPlaneVisitor, {host_plane}); in Add()
/external/tensorflow/tensorflow/core/profiler/utils/
Dgroup_events_test.cc220 XPlane* host_plane = GetOrCreateHostXPlane(&space); in TEST() local
221 XPlaneBuilder host_plane_builder(host_plane); in TEST()
242 XPlaneVisitor host_plane_visitor = CreateTfXPlaneVisitor(host_plane); in TEST()
264 XPlane* host_plane = GetOrCreateHostXPlane(&space); in TEST() local
265 XPlaneBuilder host_plane_builder(host_plane); in TEST()
289 XPlaneVisitor host_plane_visitor = CreateTfXPlaneVisitor(host_plane); in TEST()
290 const XEvent& eager_cpu_tf_op = host_plane->lines(0).events(3); in TEST()
311 XPlane* host_plane = GetOrCreateHostXPlane(&space); in TEST() local
312 XPlaneBuilder host_plane_builder(host_plane); in TEST()
343 XPlaneVisitor host_plane_visitor = CreateTfXPlaneVisitor(host_plane); in TEST()
[all …]
Dderived_timeline.cc253 XPlaneVisitor host_plane = CreateTfXPlaneVisitor(host_trace); in DeriveEventsFromHostTrace() local
254 host_plane.ForEachLine([&](const XLineVisitor& line) { in DeriveEventsFromHostTrace()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc269 const XPlane* host_plane = FindPlaneWithName(space, kCuptiDriverApiPlaneName); in TEST_F() local
270 ASSERT_NE(host_plane, nullptr); in TEST_F()
327 XPlaneVisitor host_plane = CreateTfXPlaneVisitor(cupti_host_plane); in TEST_F() local
328 EXPECT_EQ(host_plane.NumLines(), 1); in TEST_F()
340 host_plane.ForEachLine([&](const tensorflow::profiler::XLineVisitor& line) { in TEST_F()
Dcupti_collector.cc451 XPlaneBuilder* device_plane, XPlaneBuilder* host_plane) { in Flush() argument
464 auto* plane = is_host_event ? host_plane : device_plane; in Flush()
479 host_plane->ForEachLine([&](XLineBuilder line) { in Flush()
677 XPlaneBuilder host_plane( in Export() local
691 start_gpu_ns_, end_gpu_ns, &device_plane, &host_plane); in Export()
694 NormalizeTimeStamps(&host_plane, start_walltime_ns_); in Export()