Home
last modified time | relevance | path

Searched refs:device_plane (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/convert/
Dop_stats_to_tf_stats_test.cc98 XPlaneBuilder device_plane( in TEST() local
100 XLineBuilder stream1 = device_plane.GetOrCreateLine(/*line_id=*/10); in TEST()
103 &device_plane, &stream1); in TEST()
106 &device_plane, &stream1); in TEST()
107 XLineBuilder stream2 = device_plane.GetOrCreateLine(/*line_id=*/20); in TEST()
110 &device_plane, &stream2); in TEST()
113 &device_plane, &stream2); in TEST()
116 &device_plane, &stream2); in TEST()
119 /*on_device=*/true, kKernel4, kKernelDetails, &device_plane, &stream2); in TEST()
122 /*on_device=*/true, kKernel5, kKernelDetails, &device_plane, &stream2); in TEST()
Dxplane_to_trace_events_test.cc49 XPlaneBuilder device_plane(space->add_planes()); in CreateXSpace() local
50 device_plane.SetName(GpuPlaneName(0)); in CreateXSpace()
51 device_plane.SetId(0); in CreateXSpace()
52 XLineBuilder stream1 = device_plane.GetOrCreateLine(30); in CreateXSpace()
55 stream1.AddEvent(*device_plane.GetOrCreateEventMetadata("kernel1")); in CreateXSpace()
58 event3.AddStatValue(*device_plane.GetOrCreateStatMetadata("correlation id"), in CreateXSpace()
Dxplane_to_op_stats_test.cc53 XPlaneBuilder device_plane( in TEST() local
55 device_plane.AddStatValue(*device_plane.GetOrCreateStatMetadata("clock_rate"), in TEST()
57 device_plane.AddStatValue(*device_plane.GetOrCreateStatMetadata("core_count"), in TEST()
59 device_plane.AddStatValue( in TEST()
60 *device_plane.GetOrCreateStatMetadata("memory_bandwidth"), in TEST()
62 device_plane.AddStatValue( in TEST()
63 *device_plane.GetOrCreateStatMetadata("compute_cap_major"), in TEST()
65 device_plane.AddStatValue( in TEST()
66 *device_plane.GetOrCreateStatMetadata("compute_cap_minor"), in TEST()
Dxplane_to_profile_response_test.cc32 XPlaneBuilder device_plane(space->add_planes()); in CreateXSpace() local
53 device_plane.SetName("gpu:0"); in CreateXSpace()
54 device_plane.SetId(1); in CreateXSpace()
55 XLineBuilder stream1 = device_plane.GetOrCreateLine(30); in CreateXSpace()
58 stream1.AddEvent(*device_plane.GetOrCreateEventMetadata("kernel1")); in CreateXSpace()
61 event3.AddStatValue(*device_plane.GetOrCreateStatMetadata("correlation id"), in CreateXSpace()
Dxplane_to_op_metrics_db_test.cc116 XPlaneBuilder device_plane(xplane); in TEST() local
117 XLineBuilder stream1 = device_plane.GetOrCreateLine(/*line_id=*/10); in TEST()
120 &device_plane, &stream1); in TEST()
123 &device_plane, &stream1); in TEST()
124 XLineBuilder stream2 = device_plane.GetOrCreateLine(/*line_id=*/20); in TEST()
127 &device_plane, &stream2); in TEST()
130 &device_plane, &stream2); in TEST()
133 &device_plane, &stream2); in TEST()
Dxplane_to_step_events_test.cc74 XPlane* device_plane = space.add_planes(); in TEST() local
75 XPlaneBuilder device_plane_builder(device_plane); in TEST()
84 ConvertDeviceTraceXPlaneToStepEvents(*device_plane); in TEST()
Dxplane_to_op_stats.h44 DeviceCapabilities GetDeviceCapFromXPlane(const XPlane& device_plane);
51 PerfEnv GetPerfEnvFromXPlane(const XPlane& device_plane);
Dxplane_to_op_stats.cc51 DeviceCapabilities GetDeviceCapFromXPlane(const XPlane& device_plane) { in GetDeviceCapFromXPlane() argument
53 XPlaneVisitor plane = CreateTfXPlaneVisitor(&device_plane); in GetDeviceCapFromXPlane()
91 PerfEnv GetPerfEnvFromXPlane(const XPlane& device_plane) { in GetPerfEnvFromXPlane() argument
92 DeviceCapabilities cap = GetDeviceCapFromXPlane(device_plane); in GetPerfEnvFromXPlane()
Dxplane_to_trace_events.cc135 for (const XPlane* device_plane : device_planes) { in ConvertXSpaceToTraceEvents() local
136 XPlaneVisitor xplane = CreateTfXPlaneVisitor(device_plane); in ConvertXSpaceToTraceEvents()
/external/tensorflow/tensorflow/core/profiler/utils/
Dgroup_events_test.cc63 XPlane* device_plane = space.add_planes(); in TEST() local
64 XPlaneBuilder device_plane_builder(device_plane); in TEST()
75 XPlaneVisitor device_plane_visitor = CreateTfXPlaneVisitor(device_plane); in TEST()
76 EXPECT_EQ(device_plane->lines(0).events(0).stats_size(), 3); in TEST()
78 device_plane->lines(0).events(0).stats(1).metadata_id()), in TEST()
106 XPlane* device_plane = space.add_planes(); in TEST() local
107 XPlaneBuilder device_plane_builder(device_plane); in TEST()
118 XPlaneVisitor device_plane_visitor = CreateTfXPlaneVisitor(device_plane); in TEST()
119 EXPECT_EQ(device_plane->lines(0).events(0).stats_size(), 3); in TEST()
121 device_plane->lines(0).events(0).stats(1).metadata_id()), in TEST()
[all …]
Dderived_timeline.cc138 XPlaneVisitor device_plane = CreateTfXPlaneVisitor(device_trace); in DeriveEventsFromAnnotations() local
139 device_plane.ForEachLine([&](const XLineVisitor& line) { in DeriveEventsFromAnnotations()
297 XPlaneBuilder device_plane(device_traces[i]); in DeriveEventsFromHostTrace() local
299 device_plane.GetOrCreateLine(kThreadIdKernelLaunch); in DeriveEventsFromHostTrace()
307 launch_line.AddEvent(*device_plane.GetOrCreateEventMetadata( in DeriveEventsFromHostTrace()
312 device_event.AddStatValue(*device_plane.GetOrCreateStatMetadata( in DeriveEventsFromHostTrace()
316 *device_plane.GetOrCreateStatMetadata("num_launches"), in DeriveEventsFromHostTrace()
319 *device_plane.GetOrCreateStatMetadata("max_launch_time_us"), in DeriveEventsFromHostTrace()
322 *device_plane.GetOrCreateStatMetadata("avg_launch_time_us"), in DeriveEventsFromHostTrace()
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dcupti_collector.cc451 XPlaneBuilder* device_plane, XPlaneBuilder* host_plane) { in Flush() argument
464 auto* plane = is_host_event ? host_plane : device_plane; in Flush()
475 device_plane->ForEachLine([&](XLineBuilder line) { in Flush()
488 XPlaneBuilder* device_plane) { in GetDeviceCapabilities() argument
495 device_plane->AddStatValue( in GetDeviceCapabilities()
496 *device_plane->GetOrCreateStatMetadata( in GetDeviceCapabilities()
504 device_plane->AddStatValue( in GetDeviceCapabilities()
505 *device_plane->GetOrCreateStatMetadata( in GetDeviceCapabilities()
519 device_plane->AddStatValue( in GetDeviceCapabilities()
520 *device_plane->GetOrCreateStatMetadata( in GetDeviceCapabilities()
[all …]
Ddevice_tracer_test.cc272 const XPlane* device_plane = in TEST_F() local
274 ASSERT_NE(device_plane, nullptr); // Check if device plane is serialized. in TEST_F()
277 EXPECT_EQ(device_plane->event_metadata_size(), 5); in TEST_F()
279 XPlaneVisitor plane = CreateTfXPlaneVisitor(device_plane); in TEST_F()
358 XPlaneVisitor device_plane = CreateTfXPlaneVisitor(cupti_device_plane); in TEST_F() local
364 device_plane.ForEachLine([&](const tensorflow::profiler::XLineVisitor& line) { in TEST_F()