Home
last modified time | relevance | path

Searched refs:device_planes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_trace_events.cc129 std::vector<const XPlane*> device_planes = in ConvertXSpaceToTraceEvents() local
132 if (device_planes.empty()) { in ConvertXSpaceToTraceEvents()
133 device_planes = FindPlanesWithPrefix(xspace, kTpuPlanePrefix); in ConvertXSpaceToTraceEvents()
135 for (const XPlane* device_plane : device_planes) { in ConvertXSpaceToTraceEvents()
Dxplane_to_op_stats.cc151 std::vector<const XPlane*> device_planes = in ConvertXSpaceToOpStats() local
159 SetRunEnvironment(space, device_planes.size(), in ConvertXSpaceToOpStats()
166 for (const XPlane* device_trace : device_planes) { in ConvertXSpaceToOpStats()
199 bool has_device = !device_planes.empty(); in ConvertXSpaceToOpStats()